Algorithm/Programmers

[Programmers - lv01] 두 정수 사이의 합 (cpp / python)

  • -
728x90

두 정수 사이의 합

 

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

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

programmers.co.kr

문제 설명

고등학교때 배웠던 등차수열의 합 공식을 생각하면 쉬운문제다.

문제 풀이

소스코드 : C++

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

long long solution(int a, int b) {   
    return (long long)(abs(a-b)+1)*(a+b)/2;
}

소스코드 : Python

def solution(a, b):
    return (abs(a-b))*(a+b)//2

 

728x90
300x250
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.