Algorithm

[프로그래머스] 두 정수 사이의 합 - Kotlin

jwkane 2024. 4. 20. 22:30

1. 문제

 

1-1 문제 이해

매개변수 A B가 주어지면

A와 B사이에 속한 모든 정수의 합을 리턴하는 함수

 

ex) A = 5, B = 7 이라면

5,6,7 모두 합한 18 리턴 

1-2 풀이 과정

1. A부터 B더하기 / for문

for문을 이용하여 a부터 b까지 정수의 합을

answer이라는 변수에 저장한다.

class Solution {
    fun solution(a: Int, b: Int): Long {
        var answer: Long = 0
            for(i in a..b){ // A부터 B
                answer+=i // A부터 B 정수의 합을 저장
            }
        }
    }
}

2. B가 A보다 작을 땐 B부터 A 더하기 / if문

B가A보다 작을 때 A부터 B를 더하면

0을 리턴하는 것을 보았다.

 

그래서 if문을 이용해서 다른 경우의 수를

고려해보았다.

 

class Solution {
    fun solution(a: Int, b: Int): Long {
        var answer: Long = 0
        if(a<=b){ // A가 B보다 작거나 같으면
            for(i in a..b){ // A부터 B
                answer+=i
            }
        }else{ // 아니라면
            for(i in a downTo b){ // B부터 A ( A부터 B 뒷 방향)
                answer+=i
            }
        }
        return answer
    }
}