题意分析:给定长度为10的01序列,第i个数字为0代表没有重量为i的砝码,反之则代表有。利用已有砝码轮流在天平两端共放置m个砝码,要求每次满足放砝码一端重量比不放砝码一端重,dfs看每个砝码是否符合要求即可。注意连续两次放置的砝码重量不能相同。
#include<iostream>
#include<algorithm>
#include<cstring>
#include<iomanip>
#include<cctype>
#include<string>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#define LL long long
using namespace std;
const int maxn = 1100;
int ans[maxn],vis[maxn];
int pan[2],m;
char s[11];
long long a[maxn];
bool dfs(int p, int pos, int last)//当前为天平的p端,当前放置次数为pos,上一个放置的砝码重量为last
{
if(pos > m)//已放置m个砝码
return true;
int t = last;
for(int i = 1; i <= 10; i++)//遍历所有重量的砝码(如果这个重量的砝码有的话)看是否满足条件
{
if(!vis[i]) continue;
if(pan[p] + i > pan[p ^ 1] && last != i)//当前天平端重量大于另一端且当前放置砝码与上一次放置的砝码重量不相同。
{
last = i;//标记此次放置的砝码重量
pan[p] += i;//该端天平重量增加
if(dfs(p^1, pos+1, last))//遍历下一步
{
ans[pos] = i;
return true;
}
last = t;//记得要改回来看其他砝码是否满足条件。
pan[p] -= i;
}
}
return false;
}
int main()
{
memset(ans, 0, sizeof(ans));
memset(vis, 0, sizeof(vis));
memset(pan, 0, sizeof(pan));
scanf("%s",s);
for(int i = 0; i < 10; i++)
{
if(s[i] == '1') vis[i+1] = 1;//标记有无重量为i+1的砝码
}
scanf("%d",&m);
if(!dfs(0, 1, -1)) printf("NO\n");
else
{
printf("YES\n");
for(int i = 1; i <= m; i++)
{
if(i > 1) printf(" ");
printf("%d",ans[i]);
}
printf("\n");
}
return 0;
}