首页 > 代码库 > HDU 4711 Weather 概率DP

HDU 4711 Weather 概率DP

题意:有个人,他在某个区域待了n天,这个区域有m个地方,有w种天气情况,先给出这个人行程的每天的天气情况,然后给出 从第i个地方到第j个地方的概率,也可以自身到自身,然后给出 某个地方 是某种天气的概率,问你 这个人最优可能的行程路线也就是每天待在哪个地方,

概率DP,求出哪些路线概率最大 再在其中取最小字典序的

假设方程 dp[i][j] 代表 第i天待在j城市,状态转移

dp[i][j] = max(dp[i][j],dp[i - 1][k] * mp[k][j] * pp[j][nnum[i]]);其中mp[i][j]代表由i地方到j地方的概率,pp[j][nnum[i]]代表j这个地方天气为nnum[i]的概率,0<k<m,边界dp[0][0] = 1.00;

然后交了几把发现一直WA,方程没错 觉得是精度问题,因为每次有比较大小的过程,而且乘了2000次,肯定会有误差的,于是改用了long double,但是还是错误,最后还是没做出,后来看题解 发现可以用对数来解决,这高中书白读了....把mp[i][j]  跟pp[i][j]的 概率 取对数 log(mp[i][j]),log(pp[i][j]),这样后面的状态转移方程 乘法就可以写成加法了

dp[i][j] = max(dp[i][j],dp[i - 1][k] + mp[k][j] + pp[j][nnum[i]]);

然后又开始交,发现还是一直WA,因为 mp[i][j]是有可能为0的,意思是 第i个地方不可能到第j个地方,这里要判断一下把 ,若为0 则mp[i][j] = inf,当然 pp[i][j]也有可能为0,在这里判断为是否为0考虑精度问题,我是 判断它是否小于eps,  eps 我精确到了 1e-16次都不行,最后直接改成if(mp[i][j] == 0.00)反而就过了,晕死,坑点好多的题目,

int n,m,w;

int nnum[1000 + 55];

double dp[1000 + 55][100 + 55];

double mp[100 + 55][100 + 55];

double pp[100 + 55][100 + 55];

int father[1000 + 55][100 + 55];

void init() {
	memset(father,0,sizeof(father));
}

bool input() {
	while(cin>>n>>m>>w) {
		for(int i=1;i<=n;i++)
			cin>>nnum[i];
		for(int i=0;i<m;i++)
			for(int j=0;j<m;j++) {
				cin>>mp[i][j];
				if(mp[i][j] == 0.00)
					mp[i][j] = -10000000;
				else mp[i][j] = log(mp[i][j]);
			}
		for(int i=0;i<m;i++)
			for(int j=0;j<w;j++) {
				cin>>pp[i][j];
				if(mp[i][j] == 0.00)
					mp[i][j] = -10000000;
				else pp[i][j] = log(pp[i][j]);
			}
		return false;
	}
	return true;
}

void dfs(int now,int pos) {
	if(now == 1) {
		cout<<pos;
		return ;
	}
	dfs(now - 1,father[now][pos]);
	cout<<" "<<pos;
}

void cal() {
	for(int i=0;i<1000 + 55;i++)
		for(int j=0;j<100 + 55;j++)
			dp[i][j] = -10000000;
	dp[0][0] = (double)0.00;//边界
	for(int i=1;i<n + 1;i++) {
		for(int j=0;j<m;j++) {
			for(int k=0;k<m;k++) {
				//double aa = dp[i - 1][k] + mp[k][j] + pp[j][nnum[i]];
				//double bb = dp[i][j];
				//cout<<aa<<"***"<<endl;
				//cout<<bb<<"***"<<endl;
				if(dp[i - 1][k] + mp[k][j] + pp[j][nnum[i]] > dp[i][j]) {
					dp[i][j] = dp[i - 1][k] + mp[k][j] + pp[j][nnum[i]];
					father[i][j] = k;
				}
			}
		}
	}
	double p = -10000000;
	int s;
	for(int j=0;j<m;j++) {
		//cout<<dp[n][j]<<endl;
		if(dp[n][j] > p) {
			p = dp[n][j];
			s = j;
			//cout<<s<<endl;
			//printf("%.10lf\n",p);
		}
	}
	dfs(n,s);
	puts("");
}

void output() {

}

int main() {
	int t;
	cin>>t;
	while(t--) {
		init();
		if(input())return 0;
		cal();
		output();
	}
	return 0;
}



HDU 4711 Weather 概率DP