BZOJ 1802: [Ahoi2009]checker

题目描述

若有两个红格相邻
　　第一问的答案为0，所有位置上的棋子都可以通过在这两个格子上放棋子得到
　　第二设f[i]表示想让第i个格子上有棋子需要放的棋子数
若没有，第一问答案为偶数格子上白格的个数，第二问为偶数格子上红格的个数

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