能量传输

问题 E: 能量传输
时间限制: 2 Sec 内存限制: 128 MB
提交: 10 解决: 4
[提交] [状态] [命题人:admin]
题目描述
在克哈星系的冒险中，Jim和他的游骑兵遭遇了虫群的进攻。Jim出色的指挥能力使他在兵力严重不足的情况下依旧抗击虫群的多次进攻。但与此同时Jim的部分士兵战斗服能源却不足以支持他们继续作战。为了更强的火力压制，Jim决定将所有士兵的能源平分，以便所有人都能够参与作战。
已知Jim现在有n个枪兵，Jim将n个士兵围成一个圆，所有士兵的能量总和是n的倍数，且每次传输只能传输1单位的能量。更糟糕的是，充电线的长度是固定的，也就意味着，每次只能让相隔k位的两人能量交换。
休伯利安号从轨道降落需要非常长的时间，Jim希望在在最快的时间知道最少传输次数。
输入
第一行为两个正整数 n,k 表示共n个枪兵。(n<=500000,k<=n)
第二行包含n个正整数val[1]-val[n]，代表每个枪兵的能量 ( val[i]<=1e8)
保证最终结果可以用64位整型储存
输出
输出最少的传输次数。数据均保证能够平均分配所有能量。
样例输入 Copy
4 1
1 3 9 7
样例输出 Copy
6
提示
比如现在有四个人能量分别为 1 3 9 7 每次使相邻1位的两人能量交换
[1,3,9,7]→[5,3,5,7] (四次)
[5,3,5,7]→[5,5,5,5] (两次)
答案为 6次

对于 25%的数据 保证 n<=10;
对于 50%的数据 保证 n<=2000
对于 100%的数据 保证 n<=500000
不懂的可以去洛谷看糖果传递，有人疑问最后如果是偶数取中位数的话，要两个数加起来除以2，这样与两个数中随便取一个数的答案是一样的。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll s[500005];
int vis[500005];
ll he[500005];
ll p[500005];
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    ll sum=0;
    for(int i=1;i<=n;i++){scanf("%lld",&s[i]);sum+=s[i]; }
    sum/=n;
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        if(vis[i])continue;
        int j=i,len=0;
        while(!vis[j])
        {
            p[++len]=s[j];
            vis[j]=1;
           // cout<<j;
            j+=k+1;
			j=(j-1)%n+1;cout<<j<<endl;
        }
        for(int i=1;i<=len;i++)p[i]-=sum;
        for(int i=1;i<=len;i++)he[i]=he[i-1]+p[i];
      //  printf("%d\n",len);
     //   for(int i=1;i<=len;i++)printf("%lld ",he[i]);
        sort(he+1,he+1+len);
        ll e;
		 e=he[len/2+1];
        for(int ii=1;ii<=len;ii++)
        {
            ans+=labs(he[ii]-e);
        }
    }
    printf("%lld\n",ans);
}