프로그래머스, 두 정수 사이의 합(C++, Python)

프로그래머스

모든 출처는 위 사이트이다 ㅋㅋ

어렵게 꼬은 문제일 줄 알았는데 아주 아주 간단한 문제였다.

[C++]

 
#include <string>
#include <vector>
 
using namespace std;
 
long long solution(int a, int b) {
    long long answer = 0;
    long long tmp = a;
    
    if(a>b) // a가 b보다 큰 경우, 예를 들면 a=6, b=3 -> 이 경우는 swap
    {
        a = b;
        b = tmp;
    }
    
    for(;a<=b;a++)
        answer += a;
    
    return answer;
}

[Python]

def solution(a, b):
    answer = 0
    if b<a:
        tmp = a
        a = b
        b = tmp
    
    for i in range(a, b+1):
        answer = answer + i
    
    return answer