[백준] 1789번 수들의 합

c++ code

#include<iostream>
#include<vector>
using namespace std;

// 오답 원인
// 1. int -> long long
// 2. 1에 대한 예외. if(sum<S ~~~ ) -> if(sum<=S ~~~)

int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);

	long long S;
	cin >> S;
	long long idx = 1;
	while (true)
	{
		long long sum = idx * (idx + 1) / 2;
		long long sum2 = (idx + 1) * (idx + 2) / 2;
		if (sum <= S && S < sum2)break;
		idx++;
	}
	cout << idx << '\n';


	return 0;
}

 

python code

S = int(input())
n = 1
while True:
    sum = n*(n+1)/2
    sum2 = (n+1)*(n+2)/2
    if sum<=S<sum2:
        break
    else:
        n+=1
print(n)

 

result