[BOJ 27226 // C++] Лестница из чисел

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 27226번 문제인 Лестница из чисел이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/27226 

27226번: Лестница из чисел

양의 정수를 i번째 줄에 i개의 정수를 첫 줄부터 크기순으로 차례대로 작성해 만들 수 있는 삼각형의 A행부터 B행까지를 출력하는 문제이다. 단, 모든 수를 출력하는 것이 아닌 앞의 K개만을 출력해야 한다.

 

삼각형을 구성하는 A-1행의 마지막 수는 A(A-1)/2와 같음을 이용해 A행의 첫 K개의 수를 출력해줄 수 있다.

 

A행을 구성하는 수의 개수가 K보다 작을 경우 수를 K개 아닌 A개 출력해야 함에 유의하자. 또한 문제의 답이 32비트 정수 범위로 표현할 수 있는 범위를 넘을 수 있음에 유의하자.

 

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;
typedef long long ll;

ll A, B, K;

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

	cin >> A >> B >> K;
	for (ll i = A; i <= B; i++) {
		for (ll k = 1; k <= min(K, i); k++) {
			cout << i * (i - 1) / 2 + k << ' ';
		}
		cout << '\n';
	}
}