SRM 607 D1 L1：PalindromicSubstringsDiv1，DP

最新推荐文章于 2018-05-31 01:09:29 发布

xuzhezhaozhao

最新推荐文章于 2018-05-31 01:09:29 发布

阅读量1.2k

点赞数 1

分类专栏： TopCoder

本文链接：https://blog.csdn.net/xzz_hust/article/details/18988741

版权

TopCoder 专栏收录该内容

112 篇文章 0 订阅

订阅专栏

题目：http://community.topcoder.com/stat?c=problem_statement&pm=12964

需要用到概率论中求期望的数学公式：若随机变量 X = X1 + X2 + ... + Xn，则 E(X) = E(X1) + E(X2) + ... + E(Xn)。

代码：

#include <algorithm>
#include <iostream>
#include <sstream>

#include <string>
#include <vector>
#include <stack>
#include <deque>
#include <queue>
#include <set>
#include <map>

#include <cstdio>
#include <cstdlib>
#include <cctype>
#include <cmath>
#include <cstring>
#include <ctime>

using namespace std;


#define CHECKTIME() printf("%.2lf\n", (double)clock() / CLOCKS_PER_SEC)

/*************** Program Begin **********************/
double dp[5001][5001];

class PalindromicSubstringsDiv1 {
public:
    double expectedPalindromes(vector <string> S1, vector <string> S2) {
        double res = 0;
	memset(dp, 0, sizeof(dp));
	string S;
	for (int i = 0; i < S1.size(); i++) {
		S += S1[i];
	}
	for (int i = 0; i < S2.size(); i++) {
		S += S2[i];
	}
	int N = S.size();
	for (int i = 0; i < N; i++) {
		dp[i][i] = 1.0;
		res += 1.0;
	}
	for (int len = 2; len <= N; len++) {
		for (int i = 0; i <= N - len; i++) {
			int marks = 0;
			if (S[i] == '?') {
				++marks;
			}
			if (S[i + len - 1] == '?') {
				++marks;
			}
			if (2 == len) {
				if (0 == marks && S[i] == S[i + len - 1]) {
					dp[i][i + len - 1] = 1.0;
				} else if (0 != marks) {
					dp[i][i + len - 1] = 1.0 / 26.0;
				}
			} else {
				if (0 == marks && S[i] == S[i + len - 1]) {
					dp[i][i + len - 1] = dp[i+1][i + len - 2];					
				} else if (0 != marks) {
					dp[i][i + len - 1] = (1.0 / 26.0) * dp[i+1][i + len - 2];
				}
			}
			res += dp[i][i + len - 1];
		}
	}
        return res;
    }

};

/************** Program End ************************/