취미로 코딩하기

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

이번에 볼 문제는 백준 12718번 문제인 Crop Triangles (Large)이다.
문제는 아래 링크를 확인하자.

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

격자점 셋을 꼭짓점으로 갖는 삼각형의 무게중심이 다시 격자점이 되기 위해서는 세 격자점의 x좌표별 및 y좌표별 합이 각각 3의 배수가 되어야한다.

주어지는 모든 점의 x좌표와 y좌표를 각각 3으로 나눠 저장한 후 "각 좌표의 합을 3으로 나눈 나머지의 합이 다시 3의 배수"가 되게끔 세 점을 고르는 경우의 수를 계산해도 문제에서 요구하는 내용과 같은 값을 구할 수 있다.

같은 이름의 점을 중복해서 고르면 안된다는 점에 유의하여 경우의 수를 계산하자. 같은 좌표에 찍혀있더라도 점이 여러 번 주어졌다면 그 각각은 다른 점으로 취급해 계산하자.

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

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

ll sumx = 0, sumy = 0;
ll ans = 0, tmp = 1;
ll cnt[3][3];

void func(int idx) {
	if (idx == 3) {
		if (sumx % 3 == 0 && sumy % 3 == 0) ans += tmp;
	}
	else {
		for (int i = 0; i < 3; i++) {
			for (int j = 0; j < 3; j++) {
				if (cnt[i][j] > 0) {
					sumx += i, sumy += j;
					tmp *= cnt[i][j];
					cnt[i][j]--;
					func(idx + 1);
					cnt[i][j]++;
					tmp /= cnt[i][j];
					sumx -= i, sumy -= j;
				}
			}
		}
	}
}

void solve() {
	memset(cnt, 0, sizeof(cnt));
	ans = 0;
	for (int i = 0; i < 3; i++) {
		for (int j = 0; j < 3; j++) cnt[i][j] = 0;
	}

	ll N, A, B, C, D, x, y, M; cin >> N >> A >> B >> C >> D >> x >> y >> M;
	for (int i = 0; i < N; i++) {
		cnt[x % 3][y % 3]++;
		x = (A * x + B) % M;
		y = (C * y + D) % M;
	}

	func(0);

	cout << ans / 6 << '\n';
}

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

	int T; cin >> T;
	for (int t = 1; t <= T; t++) {
		cout << "Case #" << t << ": ";
		solve();
	}
}