Description
n≤100000
PS:题目好坑,描述不清
题意的目标:使得所有位置都为1.
Analysis
首先把环破坏成链。
枚举断开位置,然后贪心地运钱。
什么样的贪心策略是对的呢?
可以通过手算或者直接证明(
x2+y2≤(x+y)2
)发现,一个位置上若有0肯定用其左边的第一个非0的位置搬过来。如果搬过来那个位置变成0也无妨,那个位置再借用它左边的就行了。这样一定是最优的。
但是,可能有一些位置断开后0太多了,钱不够用了QAQ。
所以要枚举到一个合法的位置。
对于多个位置都合法,应选取最小的那个位置。
但是这样是
O(n2)
的。
事实上,只要某个位置合法,它与其他合法位置的值是一样的(挪移一下)。
所以只要找出一个合法位置并输出即可。
这样可以预处理,然后就变成
O(n)
了。
Code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
using namespace std;
typedef long long ll;
const int N=200010;
int a[N],sm[N];
int main()
{
int n,st=1;
scanf("%d",&n);
fo(i,1,n) scanf("%d",&a[i]),a[n+i]=a[i];
fo(i,1,n+n-1) sm[i]=sm[i-1]+a[i]-1;
fo(i,n,n+n-1)
if(sm[i]-sm[i-n]){st=i;break;}
for(int i=1;i<=n+n-1;i++)
{
while(sm[i]-sm[st-1]<0) st=++i;
if(i-st+1==n) break;
}
int j=st+n-1;
ll ans=0;
fd(i,st+n-1,st)
if(!a[i])
{
while(a[j]==0 || j>i) j--;
a[i]++,a[j]--,ans+=(j-i)*(j-i);
}
printf("%lld",ans);
return 0;
}