蓝桥杯：大石头的搬运工

最新推荐文章于 2024-06-10 01:05:20 发布

张小呆QAQ

最新推荐文章于 2024-06-10 01:05:20 发布

阅读量769

点赞数 2

文章标签：数据结构 c++ 算法蓝桥杯

本文链接：https://blog.csdn.net/hacker_hao/article/details/134558815

版权

思想：前缀和。先将各堆石头按照位置的先后排序。然后对于排序好的石头数组，我们遍历数组。算出假设每一个石头为中心不动，其余石头向它看齐时的花费。这个花费相当于这个中心前面的石头的移动花费+后面石头的移动花费。但是石头移动时，会将重量累加（这就是前缀和后缀和）。所以每次移动以后，要算累加后的重量 x 当前位置移动到相邻位置的距离。

代码如下：

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
#define p first
#define w second
long long pre[N], nex[N];
pair <int, int> q[N];
int main()
{
    int n;
    cin >> n;
    for(int i = 1; i <= n; i++)
        cin >> q[i].w >> q[i].p;
    sort(q+1, q+n+1);
    long long s = 0;
    for(int i = 1; i<= n;i++)
    {
        pre[i] = pre[i-1];
        pre[i] += s*(q[i].p - q[i-1].p);
        s += q[i].w;
    }
    s = 0;
    for(int i = n; i >= 1; i--)
    {
        nex[i] = nex[i+1];
        nex[i] += s*(q[i+1].p - q[i].p);
        s+=q[i].w;
    }
    long long ans = 1e18;
    for(int i = 1; i <= n; i++)
        ans = min(ans, nex[i]+pre[i]);
    cout << ans;
    return 0;

}