취미로 코딩하기

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

이번에 볼 문제는 백준 26769번 문제인 Deski이다.
문제는 아래 링크를 확인하자.

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

주어진 막대기들 중 네번째 길이의 막대를 한 변의 길이로 하는 정사각형을 만들어 문제를 해결하자.

이는 가장 큰 네 개의 변수만을 계속 저장하는 것으로 O(N)에 찾아낼 수 있다.

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

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

int N;
ll n1, n2, n3, n4;

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

	cin >> N;
	while (N--) {
		ll x; cin >> x;
		if (x >= n1) {
			n4 = n3;
			n3 = n2;
			n2 = n1;
			n1 = x;
		}
		else if (x >= n2) {
			n4 = n3;
			n3 = n2;
			n2 = x;
		}
		else if (x >= n3) {
			n4 = n3;
			n3 = x;
		}
		else if (x >= n4) n4 = x;
	}

	cout << n4 * n4;
}