题意:给你一个二维数组每个位置有一个数字这个数字代表一种颜色,然后问每种颜色的曼哈顿距离之和;
思路;我们用vector两个二维数组(X和Y)分别存储值为x这个颜色的横坐标以及纵坐标,然后我们对值为x的横纵坐标分别进行sort排序
例如X[x][i]第i个需要减去X[x][i-1],X[x][i-2]...X[x][0]
ans+(X[x][i]-X[x][i-1])+(X[x][i]-X[x][i-2])+(X[x][i]-X[x][i-3])...
也就是将X[x][i]加了i次,同时需要考虑后面的元素减去X[x][i],也就是减了size-i个X[x][i]
代码:
vector<int> X[100010],Y[100010];
void solve()
{
int n; cin>>n;
int m; cin>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
int t; cin>>t;
X[t].push_back(i);
Y[t].push_back(j);
}
int ans=0;
for(int j=1;j<=100000;j++)
{
if(X[j].size()>1)
{
sort(X[j].begin(),X[j].end());
sort(Y[j].begin(),Y[j].end());
/*
将颜色是j的横纵坐标分开,
计算第i个加上其与前面所有的差,
所以说x[i]被加了i次
x[i]也需要被后面减去(size-i)次
*/
for(int i=0;i<X[j].size();i++)
{
ans+=i*X[j][i];
ans+=i*Y[j][i];
ans-=(X[j].size()-i-1)*X[j][i];
ans-=(X[j].size()-i-1)*Y[j][i];
}
}
}
cout<<ans<<endl;
return ;
}
把一切交给时间,他会告诉我们走的路是不是对的!