[BZOJ1659][Usaco2006 Mar]Lights Out 关灯

原题地址

迭代加深搜索，从小到大枚举剩余灯泡数，然后用这个剪枝.

Mektpoy神犇的题解

把字典序最大看成最小怒WA了两发，没救了…

AC code:

#include <cstdio>
const int N=51;
const int INF=1<<30;
int n,m,lo,la=INF;
int a[N],b[N],opt[N],ans[N];

void update(){
    if(lo>la) return ;
    if(lo==la){
        for(int i=1;i<=n;i++){
            if(opt[i]>ans[i]) break;
            if(opt[i]<ans[i]) return ;
        }
    }
    la=lo;
    for(int i=1;i<=la;i++) ans[i]=opt[i];
}

bool DFS(int cnt,int pos,int k){
    if(cnt>k) return 0;
    if(pos+m-1>n){
        for(int i=pos;i<=n;i++) cnt+=a[i];
        if(cnt>k) return 0;
        update();
        return 1;
    }
    bool flag=0;
    flag|=DFS(cnt+a[pos],pos+1,k);
    opt[++lo]=pos;
    for(int i=1;i<=m;i++) a[pos+i-1]^=b[i];
    flag|=DFS(cnt+a[pos],pos+1,k);
    lo--;
    for(int i=1;i<=m;i++) a[pos+i-1]^=b[i];
    return flag;
}

int main(){
    scanf("%d%d\n",&n,&m);
    for(int i=1;i<=n;i++){
        char t;
        scanf("%c",&t);
        a[i]=t-'0';
    }
    scanf("\n");
    for(int i=1;i<=m;i++){
        char t;
        scanf("%c",&t);
        b[i]=t-'0';
    }
    for(int i=0;i<=n;i++) if(DFS(0,1,i)) break;
    printf("%d\n",la);
    for(int i=1;i<=la;i++) printf("%d\n",ans[i]);

    return 0;
}