codeforces 339C C. Xenia and Weights(dp)

题目链接：

codeforces 339C

---

题目大意：

给出砝码，砝码给定大小，是1-10中的任意个，每种砝码数量不限，问能否在轮流在天平两侧放砝码，每次保证用到的砝码和上一次不同，且当前放置砝码的一侧会因为这次放置重量超过另一侧，问能否进行m次操作，如果能，输出任意一种方案。

---

题目分析：

• 一个不算难的dp，状态定义为dp[i][j][k]代表在第i次放置采用大小为k的砝码放置后比另一侧大的重量数为j的方案是否存在。
• 转移方程就是 $$dp[i][j][k] = \exists dp[i-1][k-j][t](t \neq k )=1?1:0\ $$
• 只需要记录dp的路径，输出任意一条能够到达可行解状态的路径即可，递归倒序输出就可以了。
  

  ---

Ac代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

char s[15];
int m;
int mark[15];
int dp[1007][15][15];
int back[1007][15][15];

void print ( int i , int j , int k )
{
    if ( i != 1 )
        print ( i-1 , k-j , back[i][j][k] );
    printf ( "%d " , k );
}

int main ( )
{
    while ( ~scanf ( "%s" , s+1 ) )
    {
        scanf ( "%d" , &m );
        for ( int i = 1 ; i < 11 ; i++ )
            if ( s[i] == '1' ) mark[i] = 1;
            else mark[i] = 0;
        memset ( dp , 0 , sizeof ( dp ) );
        for ( int i = 1 ; i < 11 ; i++ )
            if ( mark[i] )
                dp[1][i][i] = 1;
        for ( int i = 2 ; i <= m ; i++ )
            for ( int j = 1 ; j < 10 ; j++ )
                for ( int k = j+1 ; k < 11 ; k++ )
                    if ( mark[k] )
                        for ( int t = 1 ; t < 11 ; t++ )
                            if ( mark[t] && t!= k )
                                if ( dp[i-1][k-j][t] ) 
                                {
                                    dp[i][j][k] = 1;
                                    back[i][j][k] = t;
                                }
        bool flag = true;
        for ( int i = 1; i < 11 && flag ; i++ )
            for ( int j = 1 ; j < 11 && flag;j++ )
                if ( dp[m][i][j] ) 
                {
                    flag = false;
                    puts ( "YES" );
                    print ( m , i , j );
                    puts ( "");
                }
        if ( flag ) puts ( "NO" );
    }
}