BZOJ 1802: [Ahoi2009]checker

題目描述

如有兩個紅格相鄰
　　第一問的答案爲0，全部位置上的棋子均可以經過在這兩個格子上放棋子獲得
　　第二設f[i]表示想讓第i個格子上有棋子須要放的棋子數
若沒有，第一問答案爲偶數格子上白格的個數，第二問爲偶數格子上紅格的個數 

#include<complex>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e3+7;
int n;
bool flag;
long long f[N];
bool a[N];
int qread()
{
    int x=0;
    char ch=getchar();
    while(ch<'0' || ch>'9')ch=getchar();
    while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        a[i]=qread();
    for(int i=3;i<=n;i++)
        if(a[i]&a[i-1])flag=1;
    if(!flag)
    {
        int b=0,c=0;
        for(int i=2;i<=n;i+=2)
            a[i]?c++:b++;
        printf("%d\n%d\n",b,c);
        return 0;
    }
    memset(f,0x3f,sizeof(f));
    f[1]=0;
    for(int i=2;i<=n;i++)
        if(a[i])f[i]=1;
    for(int i=2;i<=n;i++)
        if(a[i]&a[i+1])
        {
            for(int j=i-1;j>1;j--)
                f[j]=min(f[j],f[j+1]+f[j+2]);
            for(int j=i+2;j<=n;j++)
                f[j]=min(f[j],f[j-1]+f[j-2]);
        }
    long long ans=0;
    for(int i=2;i<=n;i+=2)
        ans+=f[i];
    printf("0\n%lld\n",ans);
    return 0;
}