[CodingTest] 두 정수 사이의 합

hye·2023년 2월 18일

CodingTest Lv.1 programmers

AlgorithmTest

목록 보기

15/95

📖 Exam

문제 설명

두 정수 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

✍ Answer

class Solution {
    public long solution(int a, int b) {
        long answer = 0;
        int i = 0;

        if(a>b){
            for(i = 0; i<(a-b); i++){
                answer += (b+i);
            }
            answer+=a;
        } else if(b>a){
            for(i = 0; i<(b-a); i++){
                answer += (a+i);
            }
            answer+=b;
        } else if(b==a) {
            return a;
        }
        return answer;
    }
}

💡 Realization

깔끔하지않는... 그저 산수로 푼 문제.

등차수열의 합

class Solution {
    public long solution(int a, int b) {
        return sumAtoB(Math.min(a, b), Math.max(b, a));
    }

    private long sumAtoB(long a, long b) {
        return (b - a + 1) * (a + b) / 2; // 등차수열의 합 공식
    }
}

for문 안에 삼항연산자

class Solution {
  public long solution(int a, int b) {
      long answer = 0;
      for (int i = ((a < b) ? a : b); i <= ((a < b) ? b : a); i++) 
          answer += i;

      return answer;
  }
}