C - Crisis at the Wedding Gym
题目链接
Gym-103185C
先说题意,有一个圆桌,客人总数为n,每位客人有数目不等的杯子,
服务员可以顺时针或逆时针走,带着k个杯子走一步的cost为k,问为了把杯子平均分配的最小cost,题目保证杯子总数能被n整除。
解题思路:
先顺时针从一个点出发,服务员走一圈,构造sum1
a[i]为i号客人的杯子数,sum1[i]为从1出发到i点到i+1点的cost,过程中sum1[i]可能出现负数,也就是i点缺杯子,要走第二圈,先暂时不理。
从其他的起点出发,可知如果在sum1[i]最小值的i点后出发,sum1[i]不会出现负数,即只走一圈
令sum为sum1数组的总和,mi为sum1数组中最小值,可知mi<=0
ans=sum-mi*n(也就是起点前少了一个mi,新的sum1数组整体都要减去mi)
这样再考虑逆时针走的情况,ans为两种情况中小的那一个
附AC代码
#include <iostream>
#include <cstdio>
#include <algorithm>
#define ll long long
#define int long long
using namespace std;
const int maxn=1e5+5;
int sum1[maxn];
int sum2[maxn];
int a[maxn];
int b[maxn];
signed main()
{
//ios::sync_with_stdio(false);
int n;
cin>>n;
int sum=0;
for(int i=1;i<=n;i++)
{
cin>>a[i];
b[n-i+1]=a[i];
sum+=a[i];
}
int av=sum/n;
sum=0;
int mi=1e9;
for(int i=1;i<=n;i++)
{
a[i]-=av;
sum1[i]=sum1[i-1]+a[i];
sum+=sum1[i];
mi=min(mi,sum1[i]);
}
int ans=sum-mi*n;
sum=0;
mi=1e9;
for(int i=1;i<=n;i++)
{
b[i]-=av;
sum2[i]=sum2[i-1]+b[i];
sum+=sum2[i];
mi=min(mi,sum2[i]);
}
ans=min(ans,sum-mi*n);
cout<<ans<<endl;
return 0;
}
一些碎碎念:
训练时这道题没想出来,有这样模糊的思路,结束后看了别人的ac代码,想了一晚上终于想通了