Xenia and Weights

 

题意：给定一个长度为10的字符串，第i位上为1表示有重量为i的砝码，然后我们需要进行m次加码，左右轮流加，要求每次加的砝码和上次不同而且加完后当前加的边应比另外一边重。

题解：数据小，直接dfs搜

#include<iostream>
#include<cstdio>
#include<map>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<stack>
#include<queue>
#include<deque>
#include<cstring>
#include<vector>
#include<set>
#include<cmath>
#include<cstring>
#include<cstdlib>
#define fi first
#define se second
#define u1 (u<<1)
#define u2 (u<<1|1)
#define PII pair<int,int>
#define ll long long
#define ull unsigned long long
#define PLL pair<long long,long long>
#define scd(a) scanf("%d",&a)
#define scdd(a,b) scanf("%d%d",&a,&b)
#define scddd(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define scl(a) scanf("%lld",&a)
#define rep(i,n) for(int i = 0; (i)<(n); i++)
#define rep1(i,n) for(int i = 1; (i)<=(n); i++)
#define pb(a) push_back(a)
#define mst(a,b) memset(a, b, sizeof a)
#define ac cout<<ans<<endl
using namespace std;
vector<int> ans;
string s;
bool dfs(int deep,int l,int r,char now)       //次数，左右重量，当前应该在哪一边添加砝码
{
    if(deep==0) return true;
    for(int i=1;i<=10;i++)
    {
        if(s[i]=='1'){
            if(now=='l'){
                if(!ans.size()||ans.size()&&ans.back()!=i&&l+i>r){
                    ans.push_back(i);
                    if(dfs(deep-1,l+i,r,'r')) return true;
                    ans.pop_back();
                }
            }
            else{
                if(!ans.size()||ans.size()&&ans.back()!=i&&r+i>l){
                    ans.push_back(i);
                    if(dfs(deep-1,l,r+i,'l')) return true;
                    ans.pop_back();
                }
            }
        }
    }
    return false;
}

void solve()
{

    cin>>s;
    int m;
    cin>>m;
    s="1"+s;
    if(dfs(m,1,1,'l')){
        cout<<"YES"<<endl;
        for(int i=0;i<ans.size();i++) cout<<ans[i]<<' ';
    }
    else cout<<"NO"<<endl;
}
int main()
{
    //init();
    int t=1;
    //scd(t);
    while(t--) solve();
    return 0;
}