bzoj 1704: [Usaco2007 Mar]Face The Right Way 自动转身机【贪心+差分】

首先O(n^3)的贪心很好想,就是枚举k然后从前往后扫,扫到反就翻转区间
然后考虑优化掉翻转区间维,就是搞成差分的形式,在翻转区间的尾部打上标记,再用一个变量维护当前的翻转次数,加到当前状态上来判断是否需要翻转即可

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=5005;
int n,a[N],b[N],ans=1e9,sum,f[N];
char s[5];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        if(s[0]=='B')
            a[i]=1;
    }
    for(int k=1;k<=n;k++)
    {
        memset(f,0,sizeof(f));
        int su=0,pd=1,t=0;
        for(int i=1;i<=n-k+1;i++)
        {
            if((a[i]+t)&1==1)
                f[i+k-1]++,t^=1,su++;
            if(f[i])
                t^=1,f[i]=0;
        }
        for(int i=n-k+2;i<=n;i++)
        {
            if((a[i]+t)&1==1)
                pd=0;
            if(f[i])
                t^=1;
            f[i]=0;
        }
        if(pd&&su<ans)
            ans=su,sum=k;
    }
    printf("%d %d\n",sum,ans);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/lokiii/p/9090121.html
今日推荐