题意很明确,给出天平的砝码种类,要求按照左右左右这样的顺序方砝码,相邻两次的砝码不能相同,且每次放完砝码的一端要更重。
dfs就行了。
增长了搜索的姿势,
if(dfs(v,2,cnt+1)) return true;
这样写dfs话,在最后一定不要忘了return。
【代码】
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<set>
#include<map>
using namespace std;
const int maxn = 10050;
typedef long long LL;
int arr[1010];
vector<int>vt;
vector<int>tt;
int l = 0,r = 0 , k ,res = 0;
int dfs(int x,int type,int cnt){
if(cnt == k) {
res = 1;return true;
}
for(int i=0;i<vt.size();i++){
int v = vt[i];
if(v == x) continue;
if(type == 1 && v + l > r){
l += v;
tt.push_back(v);
if(dfs(v,2,cnt+1)) return true;
tt.pop_back();
l -= v;
}else if(type == 2 && v + r > l){
r += v;
tt.push_back(v);
if(dfs(v,1,cnt+1)) return true;
r -= v;
tt.pop_back();
}
}
return false;
}
int main(){
string tmp;
cin>>tmp;
cin>>k;
int len = tmp.length();
for(int i=0;i<len;i++){
arr[i] = tmp[i] - '0';
if(arr[i]){
vt.push_back(i+1);
}
}
l = r = 0;
res = 0;
dfs(-1,1,0);
if(res) {
puts("YES");
for(int i=0;i<tt.size();i++){
printf("%d ",tt[i]);
}
puts("");
}
else puts("NO");
return 0;
}