洛谷P1789 【Mc生存】插火把经典解法

#include<stdio.h>
#include<stdlib.h>
#include<math.h>
int main(void)
{
	int n, m, k, i, a, b, x, y, s = 0, j;
	scanf("%d%d%d", &n, &m, &k);
	int* q = (int*)calloc((n+4) * (n+4), sizeof(int));
	q = q + (n + 6);
	for (i = 0; i < m; i++)
	{
		scanf("%d%d", &x, &y);
		for (a = -2; a <= 2; a++)
			for (b = -2; b <= 2; b++)
				if (abs(a)+abs(b) <= 2)
					*(q + x-1 + a + (y-1 + b) * (n + 4)) = 1;
	}
	for (; i < m+k; i++)
	{
		scanf("%d%d", &x, &y);
		for (a = -2; a <= 2; a++)
			for (b = -2; b <= 2; b++)
				*(q + x-1 + a + (y + b-1) * (n + 4)) = 1;
	}
	for (i = 0; i < n; i++)
		for (j = 0; j < n; j++)
			if (*(q + i*(n+4)+j) == 0)
				s++;
	printf("%d", s);
	return(0);
}