洛谷 P2512 [HAOI2008]糖果传递

题目

题目大意

有n个小朋友坐成一圈,每人有a_i颗糖果。

每人只能给左右两人传递糖果。

每人每次传递一颗糖果的代价为1。

求使所有人获得均等糖果的最小代价。

对于100%的数据,n<=10^6。

题目分析

有一..难。不要着急。

我搞明白了!

       设x[i]表示i+1向i传的糖果数,x[n]表示1向n传的糖果数

  a[1]+x[1]−x[n]=\bar{a}
  a[2]+x[2]−x[1]=\bar{a}
  a[3]+x[3]−x[2]=\bar{a}
  ⋯⋯⋯⋯
  a[n−1]+x[n−1]−x[n−2]=\bar{a}

  把式子变形:

  x[1]=\bar{a}−a[1]+x[n]
  x[2]=\bar{a}−a[2]+x[1]=2∗\bar{a}−a[2]−a[1]+x[n]
  x[3]=\bar{a}−a[3]+x[2]=3∗\bar{a}−a[3]−a[2]−a[1]+x[n]
  ⋯⋯⋯⋯
  x[n−1]=\bar{a}−a[n−1]+x[n−2]=(n−1)∗\bar{a}\sum_{i=1}^{n-1}a[i]+x[n]
  x[n]=n∗\bar{a}\sum_{i=1}^{n}a[i]+x[n]=0+x[n]\bar{}

  设s[i]=\sum_{j=1}^{i}a[j]−i∗\bar{a},则: 

     ans=∑∣x[i]∣=∑∣s[i]−x[n] ∣

  所以当x[n]为{s[1],s[2],...,s[n]}的中位数时答案最小

以上摘自:https://www.cnblogs.com/CtrlCV/p/5626194.html 蒟蒻终于看懂了/哭/!!! 

看懂了吗???如果你晕了,估计是卡在s[i]那了吧。反正我刚开始是(我也太弱了吧)。

那我添油加醋(貌似是个贬义词??)说点:(红色的是新加的辅助理解)

       设s[i]=\sum_{j=1}^{i}a[j]−i∗\bar{a}

  x[1]=\bar{a}−a[1]+x[n]=x[n]-s[1]
  x[2]=\bar{a}−a[2]+x[1]=2∗−a[2]−a[1]+x[n]=x[n]-s[2]
  x[3]=\bar{a}−a[3]+x[2]=3∗\bar{a}−a[3]−a[2]−a[1]+x[n]=x[n]-s[3]
  ⋯⋯⋯⋯
  x[n−1]=\bar{a}−a[n−1]+x[n−2]=(n−1)∗\bar{a}\sum_{i=1}^{n-1}a[i]+x[n]=x[n]-s[n-1]
  x[n]=n∗\bar{a}\sum_{i=1}^{n}a[i]+x[n]=0+x[n]=x[n]-s[n]\bar{}(话说s[n]等于0这个不用说吧)

  

     ans=∑∣x[i]∣=∑∣x[n]-s[i] ∣=∑∣s[i]−x[n] ∣

 然后最后一句:所以当x[n]为{s[1],s[2],...,s[n]}的中位数时答案最小

这个应该可以证明ba,但是我已经强调我蒟了(强行甩锅),可以百度一下??

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

long long ave,x=0,ans=0;
long long a[1000010],s[1000010];

int main()
{
    int n; scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);//每个人的初始糖果数
        ave+=a[i];//平均数ave
    }
    ave/=n; a[n+1]=a[1];
    for(int i=1;i<=n;i++)
    {
    	x+=a[i];
    	s[i]=x-i*ave;
    }
    sort(s+1,s+1+n);
    long long mid=s[n/2+1];//中位数
    for(int i=1;i<=n;i++) ans+=abs(s[i]-mid);
    printf("%lld",ans);
    return 0;
}

记得最爱你的long long

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
这道题目还可以使用树状数组或线段树来实现,时间复杂度也为 $\mathcal{O}(n\log n)$。这里给出使用树状数组的实现代码。 解题思路: 1. 读入数据; 2. 将原数列离散化,得到一个新的数列 b; 3. 从右往左依次将 b 数列中的元素插入到树状数组中,并计算逆序对数; 4. 输出逆序对数。 代码实现: ```c++ #include <cstdio> #include <cstdlib> #include <algorithm> const int MAXN = 500005; struct Node { int val, id; bool operator<(const Node& other) const { return val < other.val; } } nodes[MAXN]; int n, a[MAXN], b[MAXN], c[MAXN]; long long ans; inline int lowbit(int x) { return x & (-x); } void update(int x, int val) { for (int i = x; i <= n; i += lowbit(i)) { c[i] += val; } } int query(int x) { int res = 0; for (int i = x; i > 0; i -= lowbit(i)) { res += c[i]; } return res; } int main() { scanf("%d", &n); for (int i = 1; i <= n; ++i) { scanf("%d", &a[i]); nodes[i] = {a[i], i}; } std::sort(nodes + 1, nodes + n + 1); int cnt = 0; for (int i = 1; i <= n; ++i) { if (i == 1 || nodes[i].val != nodes[i - 1].val) { ++cnt; } b[nodes[i].id] = cnt; } for (int i = n; i >= 1; --i) { ans += query(b[i] - 1); update(b[i], 1); } printf("%lld\n", ans); return 0; } ``` 注意事项: - 在对原数列进行离散化时,需要记录每个元素在原数列中的位置,便于后面计算逆序对数; - 设树状数组的大小为 $n$,则树状数组中的下标从 $1$ 到 $n$,而不是从 $0$ 到 $n-1$; - 在计算逆序对数时,需要查询离散化后的数列中比当前元素小的元素个数,即查询 $b_i-1$ 位置上的值; - 在插入元素时,需要将离散化后的数列的元素从右往左依次插入树状数组中,而不是从左往右。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值