首页 > 代码库 > UVALive 5790 Ball Stacking DP
UVALive 5790 Ball Stacking DP
DP的方向真的很重要,这题做的时候死活想不出来,看了题解以后恍然大悟原来这么简单。
题意:
有n层堆成金字塔状的球,若你要选一个球,你必须把它上面那两个球取了,当然也可以一个不取。求选的球最大的权值和。
题解:
把原来的金字塔变换一下形式,转换成直角三角形的样子。
假如原三角形是这样的:
然后可以转化成这个样子:
这样子的话,要选择一个球,就要选择他左上的那个矩形。
这样就容易转移了,用dp[i][j]表示变换后的图形中,第 i 行取的最下面一个球是j,很明显这样下一行就只能小于j个球了。
不过,这样每层之间的转移是O(n^2)的,也许会T,然后可以用单调性来优化一下。
代码:
#include <iostream>#include <cstdio>#include <cstring>#include <cstdlib>#include <cmath>#include <algorithm>#include <string>#include <queue>#include <stack>#include <vector>#include <map>#include <set>#include <functional>#include <cctype>#include <time.h>using namespace std;const int INF = 1<<30;const int MAXN = 1111;int a[MAXN][MAXN];int sum[MAXN][MAXN];int dp[MAXN][MAXN];int n;int main() { #ifdef Phantom01 freopen("LA5790.txt", "r", stdin); #endif //Phantom01 while (scanf("%d", &n)!=EOF) { if (n==0) break; memset(dp, 0, sizeof(dp)); memset(sum, 0, sizeof(sum)); for (int i = 1; i <= n; i++) for (int j = 1; j <= i; j++) scanf("%d", &a[i-j+1][j]); for (int i = 1; i <= n; i++) for (int j = 1; j <= n-i+1; j++) sum[i][j] = a[i][j] + sum[i][j-1] + sum[i-1][j] - sum[i-1][j-1]; int ans = 0; for (int i = 1; i <= n; i++) { int best = n-i+2; for (int j = n-i+1; j > 0; j--) { if (dp[i-1][j]>dp[i-1][best]) best = j; dp[i][j] = dp[i-1][best]+sum[i][j] - sum[i-1][j]; ans = max(ans, dp[i][j]); } } printf("%d\n", ans); } return 0;}
UVALive 5790 Ball Stacking DP
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。