雷涛的小猫题解

雷涛的小猫题解

题目

雷涛同学非常的有爱心，在他的宿舍里，养着一只因为受伤被救助的小猫(当然，这样的行为是违反学生宿舍管理条例的)。在他的照顾下，小猫很快恢复了健康，并且愈发的活泼可爱了。?

可是有一天，雷涛下课回到寝室，却发现小猫不见了！经过一番寻找，才发现她正趴在阳台上对窗外的柿子树发呆…

在北京大学的校园里，有许多柿子树，在雷涛所在的宿舍楼前，就有N棵。并且这N棵柿子树每棵的高度都是H。冬天的寒冷渐渐笼罩了大地，树上的叶子渐渐掉光了，只剩下一个个黄澄澄的柿子，看着非常喜人。而雷涛的小猫恰好非常的爱吃柿子，看着窗外树上的柿子，她十分眼馋，于是决定利用自己敏捷的跳跃能力跳到树上去吃柿子。

小猫可以从宿舍的阳台上跳到窗外任意一棵柿子树的树顶。之后，她每次都可以在当前位置沿着当前所在的柿子树向下跳1单位距离。当然，小猫的能力远不止如此，她还可以在树之间跳跃。每次她都可以从当前这棵树跳到另外的任意一棵，在这个过程中，她的高度会下降Delta单位距离。每个时刻，只要她所在的位置有柿子，她就可以吃掉。整个“吃柿子行动”一直到小猫落到地面上为止。

雷涛调查了所有柿子树上柿子的生长情况。他很想知道，小猫从阳台出发，最多能吃到多少柿子？他知道写一个程序可以很容易的解决这个问题，但是他现在懒于写任何代码。于是，现在你的任务就是帮助雷涛写一个这样的程序。

图为N=3，H=10，Delta=2的一个例子。小猫按照图示路线进行跳跃，可以吃到最多的8个柿子

题目大意

每棵树的不同地方都有柿子，猫可以在所有的树之间跳跃，可以选择留在当前树上并下降高度1，或选择跳到其他树上并下降高度delta，求最多可吃到的柿子。?

基本思路

用mp记录每棵树的每个高度上的柿子数量 从上到下dp
dp[i][j]表示动态规划做到第i棵树第j高度时能吃到的最多柿子数

for(register int j=h;j>=0;j--){
	for(register int i=1;i<=n;i++){
		for(register int k=1;k<=n;k++){
			if(i==k)continue;
			dp[i][j]=max(dp[i][j+1],dp[k][j+delta]);
		}
		if(mp[i][j])dp[i][j]+=mp[i][j];
	}
}

$O(n^2\ast h)$的算法 ? 太慢了

然而可以通过jm去掉内层的k循环，jm[j]表示dp[1~n][j]的最大值

dp[i][j]=max(dp[i][j+1],jm[j+delta]);

–> 当前树的j+1的高度与其他树中(j+delta的高度)的最大值

从而优化成$O(n\ast h)$

代码实现

#include <bits/stdc++.h>
using namespace std;
const int maxn=5111;
int dp[maxn][maxn];
int n,h,delta;
int mp[maxn][maxn];
int jm[maxn]; //记录同一高度上 所有树中 对应dp值的最大值(降复杂度用)
int main(){
	memset(jm,0,sizeof jm);
	memset(dp,0,sizeof dp);
	memset(mp,0,sizeof mp);
	cin>>n>>h>>delta;
	for(int i=1;i<=n;i++){
		int op;
		cin>>op;
		for(int j=1;j<=op;j++){
			int tmp;
			cin>>tmp;
			mp[i][tmp]++;
		}
	} 
	for(register int j=h;j>=0;j--){
		for(register int i=1;i<=n;i++){
			dp[i][j]=max(dp[i][j+1],jm[j+delta]);
			if(mp[i][j])dp[i][j]+=mp[i][j];
			jm[j]=max(jm[j],dp[i][j]); 
		}
	}
	int ans=-9999999;
	for(int i=0;i<=n;i++){
		for(int j=0;j<=h;j++){
			ans=max(ans,dp[i][j]);
		}
	}
	cout<<ans<<endl;
	return 0;
}