- 首先,可以知道,棋盘上A,B两点之间的距离(九向联通)等于
max(|xA−xB|,|yA−yB|)
。
- 但是这个东西不好求和,需要对它做一个变形。
- 设
x′A=xA+yA2,y′A=xA−yA2
于是
max(|xA−xB|,|yA−yB|)=max(∣∣(x′A+y′A)−(x′B+y′B)∣∣,∣∣(x′A−y′A)−(x′B−y′B)∣∣)=max(x′A−x′B+y′A−y′B , x′A−x′B+y′B−y′A , x′B−x′A+y′A−y′B , x′B−x′A+y′B−y′A)=max(x′A−x′B , x′B−x′A)+max(y′A−y′B , y′B−y′A)=∣∣x′A−x′B∣∣+∣∣y′A−y′B∣∣
- 即曼哈顿距离。
于是我们可以把
x′,y′
排序来计算他们的贡献。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int M=1e6+5;
char str[1005][1005];
int s[2][M],d[2][M],t,k,n,m;
ll calc(int f,int n){
ll sum=0;
sort(s[f]+1,s[f]+1+n);
sort(d[f]+1,d[f]+1+n);
for(int i=1;i<=n;i++)sum+=1ll*(s[f][i]+d[f][i])*(i-(n-i+1));
printf("%lld ",sum/2);
}
int main(){
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)scanf("%s",str[i]+1);
for(int i=1;i<=n;i++)for(int j=1;j<=m;j++){
if(str[i][j]=='M')s[0][++t]=i+j,d[0][t]=i-j;
if(str[i][j]=='S')s[1][++k]=i+j,d[1][k]=i-j;
}calc(0,t);calc(1,k);
return 0;
}