알고리즘 풀이/프로그래머스

[Level1] 두 정수 사이의 합

mhko411 2021. 1. 21. 21:16
728x90

문제

두 정수 a, b가 입력되었을 때 a와 b사이의 정수를 더하여 출력해보자.

입력되는 두 개의 정수는 대소구분없이 입력되며  -10,000,000 이상 10,000,000 이하의 정수이다.

 

입력

3 5

5 3

3 3

 

출력

12

12

3


1. a와 b 사이의 크기를 비교하여 합을 구한다.

2. a와 b가 같다면 a를 return한다.

3. 이후 a와 b의 크기에 따라 다른 for문으로 합을 구하도록 한다.

 

python

def solution(a, b):
    answer = 0
    if a==b:
        return a
    elif a>b:
        for num in range(b,a+1):
            answer+=num
    else:
        for num in range(a,b+1):
            answer+=num
    return answer

 

c++

#include <string>
#include <vector>

using namespace std;

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