필요/코딩테스트(프로그래머스)

[프로그래머스/C++] ㅡ LV1 - 두 정수 사이의 합

ㅊㅇㅅ 2021. 3. 5. 18:25

문제

programmers.co.kr/learn/courses/30/lessons/12912

 

코딩테스트 연습 - 두 정수 사이의 합

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요. 예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다. 제한 조건 a와 b가 같은 경우

programmers.co.kr

풀이

#include <string>
#include <vector>

using namespace std;

long long solution(int a, int b) {
    long long answer = 0;
    if (a < b) {	// a가 적을 때 a부터 덧셈 시작
        for (int i=a; i<=b; i++) {
           answer+=i;
       }
    }
    else {	// b가 적을 때 b부터 덧셈 시작
        for (int i=b; i<=a; i++) {
           answer+=i;
       }
    }
    return answer;
}

풀이

반복문에서 반복 시키는 인수 i를 a 또는 b로 초기화 하면 된다.

a와 b의 대소관계를 고려하지 않으므로, 각각의 case를 if문으로 나눴다.