洛谷3137 USACO Circular Barn

非常烧脑的一道题目，一开始我的方法是倒着枚举0然后找最远的a[i]>0的位置，这题有个很明显的贪心结论就是如果有能转移过来的牛 就算那个点只是1 也应该转移 由后面的点补分。 这题的数据规模其实暗示了可以枚举 但是我的确想不清楚枚举的正确性…但是贪心正确性应该是明显的（我想枚举的正确性可能是 总有一个点应该是保留自己一个位置？但是说实话这个证明我没看到有人写。

code

#include <iostream>
#include <cstring>
#include <queue>
#define MAXN 5005
using namespace std;
int N;
int a[MAXN],ans=99999999;
queue<int > q;
int main()
{
 int i, j, k;
 cin>>N;
 for(i=1;i<=N;i++)
 {
  cin>>a[i];
  a[i+N]=a[i];
 }
 for(i=1;i<=N;i++)
 {
  int cnt=0,flag=0;
  while(!q.empty())
  q.pop();
  for(j=i;j<=i+N-1;j++)
  {
   if(q.empty()&&!a[j])
   {
    flag=1;
    break;
   }
   int temp=a[j];
   while(temp--)
    q.push(j);
   int t=q.front();
   q.pop();
   cnt+=(j-t)*(j-t);
  }
  if(flag)
  continue;
  ans=min(ans,cnt);
 }
 cout<<ans<<endl;
}