[level 1] 두 정수 사이의 합

문제 설명

두 정수 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 사이의 부분합을 구해야한다.

세가지 경우의 수가 있다.

  1. a = b일 경우

  2. a > b일 경우

  3. a < b일 경우

1번의 경우는 사이의 수가 없기 때문에 둘 중 하나를 그냥 리턴하면된다.

2,3번은 사이의 합을 구해야 한다.

등차수열 합 공식을 사용할까 했지만 단순반복으로 풀어보았다. (등차수열은 나중에 다시 만들어보자)

 

알고리즘

1. a와 b를 입력받음

2. 만약 a = b일 경우, 둘 중 하나를 리턴

3. 만약 a > b일 경우, b부터 a까지 반복해서 answer에 더함.

4. 만약 a < b일 경우, a부터 b까지 반복해서 answer에 더함.

5. answer을 리턴

 

풀이 코드

#include <string>
#include <vector>

using namespace std;

long long solution(int a, int b) {

    long long answer = 0;

    if (a == b)
        answer = a;

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

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

    return answer;
}

https://school.programmers.co.kr/learn/courses/30/lessons/12912

프로그래머스