解题思路:本题在输入时将同一种颜色的坐标归为一类。对于每一种颜色都按照坐标进行排序,并分别计算其两两之间的距离即可。
而计算距离时,直接两两配对的O(n^2)肯定会tle,所以这里应当利用前缀和。这里的前缀和是记录一个点在x轴或者y轴上正距离的多少。而两个点的正距离之差的绝对值就是两个点的举例了。那么如果这个点的前面有n个点,那么对于前面的所有点到这个点的距离之和,就是n个该点正距离减去前面每一个点的正距离,即减去该点之前的前缀和。
AC代码:
#include<bits/stdc++.h>
using namespace std;
long long m,n,ans=0,maxx=0,len[100010];//maxx记录最大的那个颜色数字,len记录每个颜色有几个点,颜色即下标
vector<long long> x[100010];//记录不同颜色的点的x坐标,同一个颜色的下标相同
vector<long long> y[100010];//记录不同颜色的点的y坐标,同一个颜色的下标相同
int main(){
cin>>n>>m;
for(long long i=1;i<=n;i++){//输入部分
for(long long j=1;j<=m;j++){
long long buffer;
scanf("%lld",&buffer);
maxx=max(maxx,buffer);//如果当前颜色数字大于maxx,则更新
x[buffer].push_back(i);
y[buffer].push_back(j);
len[buffer]++;//改颜色点数量加一
}
}
for(int i=1;i<=maxx;i++){//处理部分
if(len[i]<=1) continue;//该颜色没点或者只有一个点则不产生距离
long long sum[100010];//用于储存前缀和
sort(x[i].begin(),x[i].end());//排序x
sum[0]=x[i][0];//x轴上的距离前缀和初值
for(int j=1;j<len[i];j++){
sum[j]=sum[j-1]+x[i][j];//利用前缀和计算距离
ans+=j*x[i][j]-sum[j-1];
}
sort(y[i].begin(),y[i].end());//排序y
sum[0]=y[i][0];//y轴上的距离前缀和初值
for(int j=1;j<len[i];j++){
sum[j]=sum[j-1]+y[i][j];//利用前缀和计算距离
ans+=j*y[i][j]-sum[j-1];
}
}
cout<<ans;
return 0;
}