码迷,mamicode.com
首页 > 其他好文 > 详细

code vs 1029 遍历问题 区间dp

时间:2015-06-14 22:40:20      阅读:153      评论:0      收藏:0      [点我收藏+]

标签:

http://codevs.cn/problem/1029/

给出一棵二叉树(节点是小写字符)的按照先序遍历和后续遍历得到的字符串,其实就是求有多少和二叉树的先序遍历和后序遍历满足这两个字符串。
区间dp:dp(l, r, a, b)表示s字符串的(l, r)段和t字符串的(a, b)段相匹配的方案数。那么s[l]和t[b]必须一样,因为这两个是这一段的根节点。然后我们再枚举(l,r)的左子树(l+1,k)和(a,b)的左子树(a,a+k-l-1);
dp(l,r,a,b) = sum(dp(l+1,k,a,a+k-l-1)*dp(k+1,r,a+k-l,b-1));
注意还要加上仅有一棵子树的情况dp(l+1,r,a,b-1),这种情况是在左子树和右子树都是同样的所以dp(l,r,a,b)+=dp(l+1,r,a,b-1)*2;然后开心的记忆化吧。

 

技术分享
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

string _a, _b;

long long dp[30][30][30][30];

long long DP(int l, int r, int a, int b) { // cout << l << " " << r << " " << a << " " << b << endl;
    // if (l > r) return 1;

    if (dp[l][r][a][b] != -1) return dp[l][r][a][b];

    if (_a[l] != _b[b]) return dp[l][r][a][b] = 0;

    if (l == r ) return dp[l][r][a][b] = 1;

    long long res = DP(l+1, r, a, b-1) * 2;
    
    for (int k=l+1; k<r; k++) {
        res += DP(l+1, k, a, a+k-l-1) * DP(k+1, r, a+k-l, b-1);
    }
    return dp[l][r][a][b] = res;
}

int main () {
    cin >> _a >> _b;

    memset(dp, -1, sizeof(dp));
    cout << DP(0, _a.size()-1, 0, _b.size()-1) << endl;
    return 0;
}
View Code

 

code vs 1029 遍历问题 区间dp

标签:

原文地址:http://www.cnblogs.com/xuelanghu/p/4575779.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!