南昌邀请赛网络赛 D.Match Stick Game（dp）

Posted 2021-02-17 heyuhhh

南昌邀请赛网络赛 D.Match Stick Game

题目传送门

题目就会给你一个长度为n的字符串，其中(1<n<100)。这个字符串是一个表达式，只有加减运算符，然后输入的每一个字符都是可以由若干个火柴棒拼接而成的。
现在在不改变每个数的位数，数的总数以及运算符的个数的前提下，可以对火柴棒重新拼接。询问最后可以拼接出来的最大值是多少。
这个自己看下题目可能要清楚一些= =

?
每一个字符都是由若干个火柴棒构成的，我们可以考虑类似于背包的思路来求解。
因为每个数的位数最后都没发生变化，所以我们可以预处理出(f[i][j])以及(g[i][j])，分别表示(i)位数由(j)根火柴构成的最大/最小值。
因为这里除开火柴棒个数之外还涉及到了加减号，所以我们定义(dp(i,j,0/1))为前(i)个数字，用了(j)根火柴棒，并且当前这个数字前面是(-)还是(+)。
由于数据范围比较小，所以考虑加和减两种情况进行合理转移就行了。
详见代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1005;
int n, T;
char s[N] ;
int len[N] ;
ll dp[105][N][2], f[105][N], g[105][N];
int trans1[10] = {0, 0, 1, 7, 4, 5, 9, 8} ;
int trans2[10] = {0, 0, 1, 7, 4, 2, 0, 8} ;
int num ;
int main() {
    cin >> T;
    for(int i = 1; i <= 10; i++)
        for(int j = 2; j < N; j++) {
            g[i][j] = 1e14;
            f[i][j] = f[i][j - 1] ;
            for(int k = 2; k <= 7; k++) {
                if(j - k >= 0) f[i][j] = max(f[i - 1][j - k] * 10 + trans1[k], f[i][j]) ;
            }
        }
    for(int i = 1; i <= 10; i++)
        for(int j = 2; j < N; j++) {
            if(i == 1) {
                g[i][j] = 1;
                continue ;
            }
            g[i][j] = g[i][j - 1] ;
            for(int k = 2; k <= 7; k++) {
                if(j - k >= 0) g[i][j] = min(1ll * g[i - 1][j - k] * 10 + 1ll * trans2[k], g[i][j]) ;
            }
        }
    while(T--) {
        memset(dp, 0, sizeof(dp)) ;
        scanf("%d", &n);
        scanf("%s", s + 1);
        int cnt = 0, x = 0, num = 0;
        for(int i = 1; i <= n; i++) {
            if(s[i] == '+' || s[i] == '-') {
                len[++num] = cnt ;
                cnt = 0;
                if(s[i] == '+') x += 2;
                else x += 1;
            } else {
                cnt++;
                if(s[i] == '1') x += 2 ;
                else if(s[i] == '7') x += 3 ;
                else if(s[i] == '4') x += 4 ;
                else if(s[i] == '5' || s[i] == '2' || s[i] == '3') x += 5;
                else if(s[i] == '0' || s[i] == '6' || s[i] == '9') x += 6;
                else x += 7;
            }
        }
        len[++num] = cnt;
        for(int i = 2; i <= x; i++) dp[1][i][0] = dp[1][i][1] = f[len[1]][i] ;
        for(int i = 2; i <= num; i++) {
            for(int j = 2; j <= x; j++) {
                for(int k = 2; k <= j; k++) {
                    if(j - k - 2 >= 2) {
                        dp[i][j][1] = max(dp[i][j][1], dp[i - 1][j - k - 2][1] + f[len[i]][k]) ;
                        dp[i][j][1] = max(dp[i][j][1], dp[i - 1][j - k - 2][0] + f[len[i]][k]) ;
                    }
                    if(j - k - 1 >= 2) {
                        dp[i][j][0] = max(dp[i][j][0], dp[i - 1][j - k - 1][0] - g[len[i]][k]) ;
                        dp[i][j][0] = max(dp[i][j][0], dp[i - 1][j - k - 1][1] - g[len[i]][k]) ;

                    }
                }
            }
        }
        ll ans = 0;
        ans = max(ans, max(dp[num][x][0], dp[num][x][1])) ;
        cout << ans << '
';
    }
    return 0 ;
}