洛谷 P9582 题解

题目大意

有一个 $n$ 行 $m$ 列的方格，求每一个方格与它不相邻且相同的数的个数之和。

题目解法

能通过测试点 $1\sim 3$ 的做法很容易，每对于每个数都循环扫一遍就行。

很显然，那样的做法会超时，所以要考虑优化。

优化的方法十分简单，就是运用桶思想来解决。

我们定义一个数组作为桶，在读入时顺便统计每个数出现的次数。

之后再扫描格子中的每一个数，扫描时它不相邻且相同的数的个数就是当前数出现的次数减去它自己再减去与它相邻且相等的数的个数。

最后注意：要注意边界问题，数组最好是从下标 $1$ 开始用，数组大小要注意（我比赛时就是因为数组少开了一个导致 RE 了 $3$ 个点，痛失 $30$ 分）。

参考代码

#include<iostream>
using namespace std;
int a[2002][2002];
int book[10];
int main()
{
	long long n,m,ans=0;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			scanf("%d",&a[i][j]),book[a[i][j]]++;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++){
			long long cnt=0;
			if(a[i][j+1]==a[i][j]) cnt++;
			if(a[i][j-1]==a[i][j]) cnt++;
			if(a[i+1][j]==a[i][j]) cnt++;
			if(a[i-1][j]==a[i][j]) cnt++;
			ans+=(long long)(book[a[i][j]]-cnt-1);
		}
	printf("%lld",ans);
}