Dream City

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3211

省赛的题目，用了最大权值匹配去做，TLE，没有发现它们的先后关系：如果已经从n种树中选出m种，怎么排列使答案最优，贪心，按增幅从小到大排列，有了这个结论，状态转移方程就非常好写了，

#include <iostream> using namespace std ; struct CoinTree { int increasement , coins ; } ; CoinTree C[300] ; int dp[300][300] , n , m ;//dp[i][j]表示在第i + 1天砍掉第j + 1棵树的最优值 int Cmp (const void *e1 , const void *e2) { return ((CoinTree *)e1)->increasement - ((CoinTree *)e2)->increasement ; } void Readin () { int i ; scanf ("%d%d" , &n , &m) ; for (i = 0 ; i < n ; i ++) scanf ("%d" , &C[i].coins) ; for (i = 0 ; i < n ; i ++) scanf ("%d" , &C[i].increasement) ; qsort (C , n , sizeof (CoinTree) , Cmp) ; } void Comp () { int i , j , most[300] , ans ; for (j = 0 ; j < n ; j ++) dp[0][j] = C[j].coins ; for (most[0] = dp[0][0] , j = 1 ; j < n ; j ++) most[j] = dp[0][j] > most[j - 1] ? dp[0][j] : most[j - 1] ; for (i = 1 ; i < m ; i ++) { for (j = i ; j < n ; j ++) dp[i][j] = most[j - 1] + C[j].coins + i * C[j].increasement ; most[i] = dp[i][i] ; for (j = i + 1 ; j < n ; j ++) most[j] = dp[i][j] > most[j - 1] ? dp[i][j] : most[j - 1] ; } for (ans = 0 , j = m - 1 ; j < n ; j ++) if (dp[m - 1][j] > ans) ans = dp[m - 1][j] ; printf ("%d/n" , ans) ; } int main () { int T ; scanf ("%d" , &T) ; while (T --) { Readin () ; Comp () ; } return 0 ; }