pku2785.cpp

来自「这是ACM 方面的资料 是PKU的 北京大学的出来的」· C++ 代码 · 共 81 行

#include <stdio.h>
#include <stdlib.h>

int a[16000001], b[16000001];
int N, SZ;
int nu[4001][4];

int cp(const void *a, const void *b)
{
	return *(int *)b- *(int *)a;
}

int calc()
{
	int i, j, sa, sb, sum = 0, fd, p, aa, bb;
	for (i = 0; i < N; i++)
	{
		for (j = 0; j < N; j++)
		{
			a[i * N + j] = nu[i][0] + nu[j][1];
			b[i * N + j] = 0 - nu[i][2] - nu[j][3];
		}
	}
	qsort(a, SZ, sizeof(a[0]), cp);
	qsort(b, SZ, sizeof(b[0]), cp);
	for (i = 0, j = 0; i < SZ && j < SZ; )
	{
		fd = 0;
		while (i < SZ && j < SZ)
		{
			aa = a[i];
			bb = b[j];
			if (aa > bb)
			{
				i++;
			}
			else if (aa < bb)
			{
				j++;
			}
			else
			{
				fd = 1;
				break;
			}
		}
		if (fd == 0)
		{
			break;
		}
		p = a[i];
		sa = 0, sb = 0;
		while (i < SZ && p == a[i])
		{
			sa++;
			i++;
		}
		while (j < SZ && p == b[j])
		{
			sb++;
			j++;
		}
		sum += sa * sb;
	}
	return sum;
}

int main()
{
	int i;
	scanf("%d", &N);
	SZ = N * N;
	for (i = 0; i < N; i++)
	{
		scanf("%d %d %d %d", &nu[i][0], &nu[i][1], &nu[i][2], &nu[i][3]);
	}
	printf("%d\n", calc());
	return 0;
}