蓝桥杯题目-大石头的搬运工

链接

大石头的搬运工 - 蓝桥云课 (lanqiao.cn)

知识点

前缀和，枚举

思路

        首先，我们需要明白这个问题中，每一块移动的石头的位置，并不影响后续石头的移动，也就是说，无论我们怎么移动，最后的费用只依赖于每个石头的最后的位置，与移动顺序无关。这一性质，使我们可以逐一考虑每一块石头的最后位置，并比较得出最小的总费用。

        然后，我们需要分析一下如何计算每一种放置石头的方式的总费用。一种直观的方法是，对于每个石头，我们都计算它被移动到目标位置的费用，然后将这些费用加总。但这样的计算方式在本题的数据范围下是无法接受的。我们需要寻找一种更优的方法。

       这里，我们可以运用前缀和的思想。考虑到石头移动的费用与石头的位置和距离有关，我们可以先将石头按位置排序，然后计算每个石头移动到任意位置的费用，再利用前缀和的方法将这些费用累加起来。具体地，我们可以定义两个数组pre和nex,其中pre[i]表示前i个石头都移动到第i个石头的位置的总费用，nex[i]表示第i个石头之后的所有石头都移动到第i个石头的位置的总费用。这样，对于每个石头，我们就可以算出所有石头都移动到它的位置的总费用，然后再取最小值。

代码

#include<bits/stdc++.h>
using namespace std;

using ll = long long;
const int N = 1e5+10;
pair<int,int>p[N];
ll pre[N],nex[N];

int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>p[i].second>>p[i].first;
	}
	sort(p+1,p+n+1);
	ll s = 0;
	for(int i = 1;i<=n;i++){  //计算前i块石头移动到第i块石头的总费用
		pre[i] = pre[i-1] + s * (p[i].first - p[i-1].first);
		s += p[i].second;
	}
	s = 0;
	for(int i=n;i>=1;i--){  //计算后i块石头移动到第i块石头的总费用
		nex[i] = nex[i+1] + s * (p[i+1].first - p[i].first);
		s += p[i].second;
	}
	ll ans = 1e18;
	for(int i=1;i<=n;i++){
		ans = min(ans,pre[i]+nex[i]);
	}
	cout<<ans<<endl;	
	return 0;
}