cf C. Xenia and Weights

http://codeforces.com/contest/339/problem/C

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int dp[1002][15][15];
char str[100];
int f[100];
int m;
void dfs(int i,int j,int k)
{
    if(i==0) return ;
    dfs(i-1,k-j,dp[i][j][k]);
    printf("%d ",k);
}

int main()
{
    scanf("%s",str);
    int cnt=0;
    for(int i=0; i<10; i++)
    {
        if(str[i]=='1')
        {
            f[++cnt]=i+1;
        }
    }
    scanf("%d",&m);
    memset(dp,-1,sizeof(dp));
    dp[0][0][0]=0;
    for(int i=1; i<=m; i++)
    {
        for(int j=0; j<=10; j++)
        {
            for(int k=0; k<=10; k++)
            {
                if(dp[i-1][j][k]!=-1)
                {
                    for(int c=1; c<=cnt; c++)
                    {
                        if(f[c]>j&&f[c]!=k)
                        dp[i][f[c]-j][f[c]]=k;
                    }
                }
            }
        }
    }
    for(int i=1; i<=10; i++)
    {
        for(int j=1; j<=10; j++)
        {
            if(dp[m][i][j]!=-1)
            {
                printf("YES
");
                dfs(m,i,j);
                return 0;
            }
        }
    }
    printf("NO
");
    return 0;
}