BZOJ 1659: [Usaco2006 Mar]Lights Out 关灯 迭代加深搜索

1659: [Usaco2006 Mar]Lights Out 关灯

Time Limit: 5 Sec Memory Limit: 64 MB
Submit: 320 Solved: 84
[Submit][Status][Discuss]

Description

奶牛们喜欢在黑暗中睡觉。每天晚上，他们的牲口棚有L（3<=L<=50）盏灯，他们想让亮着的灯尽可能的少。他们知道按钮开关的位置，但喜闻乐见的是他们并没有手指。你得到了一个长度为T（1<=T<=7）的插槽用以帮助奶牛们改变灯的状态。

Input

第一行，两个整数L和T。第二行给出一个长度为L的01串表示初始灯的状态，0表示灯是灭的，1表示灯是亮的。第三行给出一个长度为T的01串，表示你获得的插槽。

Output

第一行输出一个整数K，表示在满足亮着的灯最少的情况下，你要用插槽操作的次数。第二行到第K+1行，每行一个整数表示你的插槽使用的位置。
“K最小的解，并且满足解的字典序最大（即按钮开关的位置尽可能靠后）”

Sample Input

10 4

1111111111

1101

Sample Output

5

1

3

4

6

7

HINT

使用5次插槽

1111111111 初始状态

0010111111 对第一个位置使用插槽

0001101111 对第三个位置使用插槽

0000000111 对第四个位置使用插槽

0000011101 对第六个位置使用插槽

0000010000 对第七个位置使用插槽

可以证明这是满足字典序最小的最优解。

Source

Silver

题解：
枚举剩下的灯的个数，如果留下这么多盏灯是可行的，就break掉，输出解。
我之前的想法是每进行一次操作后都统计剩下的灯的个数，很慢，后来发现别人是判断搜索到的这个位置是否有灯，然后加入答案中，就会快很多。然后随便剪下枝就可以了。
由于要输出方案数最小的，所以从小到大枚举，用个中间数组存，更新最小次数的时候顺便更新答案数组。
还有一个细节，之前太着急了，每搜到一个答案就直接返回，如果是只是问最少留几盏灯这样是可行的，但这样未必能保证次数最少，所以不能立刻返回，还是要老老实实计算完。
其实时间复杂度蛮玄学的。。不过bzoj上能过就是了。

#include<bits/stdc++.h>
using namespace std;

const int N = 60;
const int M = 10;
const int inf = 0x73f3f3f;

int n,t;
int a[N],b[N],c[N];
int ans,step[N];

bool dfs(int pos,int ti,int light,int limit){
    if(light>limit) return false;
    if(ti>ans) return false;    
    if(pos>n-t+1){
        for(int i=pos;i<=n;++i) if(a[i]) ++light;
        if(light>limit) return false;
        for(int i=0;i<ti;++i) step[i]=c[i];
        ans=ti;return true;
    }
    for(int j=pos;j<=pos+t-1;++j) a[j]^=b[j-pos+1];
    c[ti]=pos;
    bool tmp=dfs(pos+1,ti+1,light+a[pos],limit);
//  if(tmp) return true;
//不能直接返回，虽然这个地方有解，但不选这个会有更优的情况,即次数更少 
    for(int j=pos;j<=pos+t-1;++j) a[j]^=b[j-pos+1];
    tmp|=dfs(pos+1,ti,light+a[pos],limit);
    return tmp;
}

int main(){
    scanf("%d%d",&n,&t);
    for(int i=1;i<=n;++i) scanf("%1d",&a[i]);
    for(int i=1;i<=t;++i) scanf("%1d",&b[i]);
    ans=inf;
    for(int i=0;i<=n;++i){
        if(dfs(1,0,0,i)) {
            printf("%d\n",ans);
            for(int j=0;j<ans;++j) printf("%d\n",step[j]);
            break;
        }
    }
    return 0;
}