Programmers/Level 1

C++ / 프로그래머스 / 두 정수 사이의 합

GitHubSeob 2021. 9. 6. 01:39

문제

https://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

문제풀이

a, b 중 적은 수부터 큰 수까지의 합을 return 하면 된다.

만약 a>b이면 swap을 하여 값을 바꾼다.

a부터 b까지의 합을 answer에 더하고 return 한다.

 

코드

#include <string>
#include <algorithm>

using namespace std;

long long solution(int a, int b) {
    long long answer = 0;
    int idx=0;
    if(a>b) swap(a,b);   
    for(idx=a;idx<=b;++idx)
        answer+=idx;
    return answer;
}