728x90
https://school.programmers.co.kr/learn/courses/30/lessons/12912?language=swift
문제 설명
두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.
제한 조건
- a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
- a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
- a와 b의 대소관계는 정해져있지 않습니다.
입출력 예
a | b | return |
---|---|---|
3 | 5 | 12 |
3 | 3 | 3 |
5 | 3 | 12 |
코드
func solution(_ a:Int, _ b:Int) -> Int64 {
var result = 0
var start = 0
var end = 0
if a == b { return Int64(a) }
if a > b {
start = b
end = a
} else {
start = a
end = b
}
for i in start ... end {
result += i
}
return Int64(result)
}
다른 사람 풀이
func solution(_ a:Int, _ b:Int) -> Int64 {
return Int64(Array(a > b ? b...a : a...b).reduce(0, +))
}
Array로 풀 수 있다는 생각은 정말 1도 못했는데 너무 깔끔
728x90
'알고리즘 > 프로그래머스' 카테고리의 다른 글
프로그래머스: 연습문제 - 문자열 내림차순으로 배치하기 (0) | 2022.08.11 |
---|---|
프로그래머스: 연습문제 - 문자열 내 p와 y의 개수 (0) | 2022.08.10 |
프로그래머스: 연습문제 - 나누어 떨어지는 숫자 배열 (0) | 2022.08.08 |
프로그래머스: 연습문제 - 가운데 글자 가져오기 (0) | 2022.08.08 |
프로그래머스: 월간 코드 챌린지 시즌3 - 나머지가 1이 되는 수 찾기 (0) | 2022.08.05 |
댓글