BZOJ 3749 [POI2015]Łasuchy

Description
圆桌上摆放着n份食物，围成一圈，第i份食物所含热量为c[i]。
相邻两份食物之间坐着一个人，共有n个人。每个人有两种选择，吃自己左边或者右边的食物。如果两个人选择了同一份食物，这两个人会平分这份食物，每人获得一半的热量。
假如某个人改变自己的选择后（其他n-1个人的选择不变），可以使自己获得比原先更多的热量，那么这个人会不满意。
请你给每个人指定应该吃哪一份食物，使得所有人都能够满意。

---

【题目分析】
动态规划好题。

---

【代码】

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int f[1000002][5];//1 给左边吃;2 给右边吃;3 两个一起吃;4 两个人都不吃 
int a[1000002],n;
int ans[1000002];
bool dp(int s)
{
    memset(f,0,sizeof f);
    f[1][s]=1;
    for (int i=2;i<=n;++i)
    {
        if (f[i-1][1]&&a[i-1]<=a[i]*2) f[i][1]=1;
        if (f[i-1][4]&&a[i-1]<=a[i])   f[i][1]=4;
        if (f[i-1][2]&&a[i]<=a[i-1]*2) f[i][2]=2;
        if (f[i-1][3]&&a[i-1]>=a[i])   f[i][2]=3;
        if (f[i-1][1]&&a[i-1]<=a[i])   f[i][3]=1;
        if (f[i-1][4]&&a[i-1]*2<=a[i]) f[i][3]=4;
        if (f[i-1][2]&&a[i-1]>=a[i])   f[i][4]=2;
        if (f[i-1][3]&&a[i]*2<=a[i-1]) f[i][4]=3;
    }
    return f[n][s];
}
void print(int u)
{
    for(int i=n;i>=1;i--){
        if(u==1)ans[i-1]=(i-1)%(n-1)+1;
        if(u==2)ans[i]=(i-1)%(n-1)+1;
        if(u==3)ans[i-1]=ans[i]=(i-1)%(n-1)+1;
        u=f[i][u];
    }
    for(int i=1;i<n;i++)printf("%d%c",ans[i],i==n-1?'\n':' ');
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;++i) scanf("%d",&a[i]);
    a[++n]=a[1];
    for (int i=1;i<=4;++i)
        if (dp(i))
        {
            print(i);
            return 0;
        }
    printf("NIE\n");
    return 0;
}