취미로 코딩하기

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

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

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

어떤 하나의 색으로 칠해진 행이 $k$개라면, 국기를 회전시켜 얻은 모양에서 그 색으로 칠해져야 하는 열의 개수는 $k$개가 된다. 따라서 이 색으로 칠해진 칸 중 색을 다시 칠할 필요가 없는 칸은 $k^2$개가 된다. 또한 그 외의 해당 색으로 칠해진 칸은 모두 색을 다시 칠해야 한다.

각각의 색에 대하여 그 색이 칠해진 행의 개수를 이용해 색을 다시 칠할 필요가 없는 칸의 개수를 구하고, 전체 칸의 개수에서 이 값을 빼 문제를 해결하자.

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

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

ll N; ll ans, cnt; int old;
int A[200000];

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

	cin >> N;
	for (int i = 0; i < N; i++) cin >> A[i];
	sort(A, A + N);
	for (int i = 0; i < N; i++) {
		if (old == A[i]) cnt++;
		else {
			ans += cnt * cnt;
			cnt = 1;
			old = A[i];
		}
	}
	ans += cnt * cnt;
	cout << N * N - ans;
}