[海豹海边爆]CODEFORCE1649

原题链接:Problem - C - Codeforces

解题思路:本题在输入时将同一种颜色的坐标归为一类。对于每一种颜色都按照坐标进行排序,并分别计算其两两之间的距离即可。

而计算距离时,直接两两配对的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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值