글 읽기 - '틀렸습니다'를 '맞았습니다'로 바꾸고 싶어요

piri007 5년 전 0

두 개 두 개 짝지어서 각각 더한 후 그 두 수가 같은 경우의 수를 세었습니다. 또한

0 0 0 0

등도 고려했습니다.

반례 좀 찾아주세요.

#include <stdio.h>
#include <algorithm>

using namespace std;

int t1[16000000], t2[16000000];

int main()
{
	int n, x[4000][4];
	scanf("%d", &n);
	for (int i = 0; i < n; i++)
		scanf("%d %d %d %d", &x[i][0], &x[i][1], &x[i][2], &x[i][3]);

	for (int i = 0; i < n; i++)
	for (int j = 0; j < n; j++)
		t1[(i * n) + j] = x[i][0] + x[j][1];

	for (int i = 0; i < n; i++)
	for (int j = 0; j < n; j++)
		t2[(i * n) + j] = x[i][2] + x[j][3];

	sort(t1, t1 + (n*n));
	sort(t2, t2 + (n*n));

	int two = 0, one = n * n - 1;
	long long res = 0;
	while (one >= 0 && two < n * n)
	{
		if (-t1[one] == t2[two])
		{
			int ao = 0, at = 0;
			int mem = t2[two];
			while ((-t1[one] == mem) && one >= 0)
				ao++, one--;

			while (t2[two] == mem && two < n * n)
				at++, two++;

			res += ao * at;
		}
		else if (-t1[one] < t2[two])
			one--;
		else
			two++;
	}
	printf("%lld", res);
}