B-这是dp题吗(动态规划)
题目描述
链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
小l同学非常喜欢三角形,现在他想出个题目给小R同学做,想要考考小R同学。
题目是这样的,给你一个数字三角形,第一行1个数字,第二行3个数字,第i行2∗i−1个数字,一共n行。如下图所示,为3行的数字三角形,一开始从顶点出发,当位于某一点时可以往这个点的正下方走,也可以往左下方和右下方走。
例如在值为3的这一个点,可以往正下方走到7,也可以往左下方走到6,也可往右下方走到8(不能从3走到5或9)。
每走到一个位置就会取出当前所在位置的值,问从顶点走到最后一层,最终的往左下方和往右下方的次数之差的绝对值不能超过k次,所取的数值总和的最大值为多少。
输入描述
输出描述
示例
输入
3 1
1
2 3 4
5 6 7 8 9
输出
13
说明
第一次在值为1的点,第二次向右下方走到值为4,第三次从4这个点往正下方走到值为8的点,总和为13
思路
很明显用动态规划,此时我们要写出状态方程,因为可以往下走,往右走,往左走,反过来就是从正上方往下走、上左方往下右方走、上右方往下左方走到该点,设状态方程为f[i][j],因为要找最大值,所以方程如下:
f[i][j]=max(f[i-1][j],f[i][j]) 正上方往下走
f[i][j]=max(f[i-1][j-1],f[i][j]) 上左方往下右方走
f[i][j]=max(f[i-1][j+1],f[i][j]) 上右方往下左方走
然后我们要找的是走到第n行的最大值,但是有一个限制条件,就是最终的往左下方和往右下方的次数之差的绝对值不能超过k次,那么我们先假设最后一层的中间点是mid,那么根据题意往左或往右走次数之差不超过k次,那么走到最后一层,他的落脚点一定在[mid−k,mid+k]之间,那么问的就是走到最后一层在区间[mid−k,mid+k]之间最大值是多少。
然后按照上面的状态方程,我们存数组的时候是按三角形形状存的,那具体代码怎么写呢?
代码
#include<bits/stdc++.h>
#define int long long
using namespace std;
int dp[1005][1005],a[1005][1005];
signed main()
{
int n,k,ans=-1e18; //设置为-1e18是因为x的范围(-2e9~2e9)
cin>>n>>k;
memset(dp,-0x3f,sizeof dp); //初始化为最小,因为是找最大值并且x可能为负数
for(int i=1;i<=n;i++)
for(int j=n-i+1;j<=n+i-1;j++) //按三角形存数组,那么中间(mid)那个坐标就是n
cin>>a[i][j];
dp[1][n]=a[1][n]; //第一行肯定就是a[1][n]
for(int i=2;i<=n;i++)
for(int j=n-i+1;j<=n+i-1;j++){
dp[i][j]=max(dp[i][j],max(dp[i-1][j-1]+a[i][j],max(dp[i-1][j]+a[i][j],dp[i-1][j+1]+a[i][j]))); //状态方程找最大值,可以分开写,只是我全写在一起了,实际上是一样的
}
for(int i=n-k;i<=n+k;i++) //mid是n
ans=max(ans,dp[n][i]);
cout<<ans;
}
H-左右横跳(动态规划)
题目描述
链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
小D想和你玩个游戏,游戏规则如下:
给出 m 个高度为 n 的柱子,每个柱子的每一单位长度上有一定的分数 aij 表示在第 j 个柱子上的第 i 个单位长度上的分数,玩家可以选择从任意一个柱子上单位长度为 1 处开始,有以下两种操作:
(1)向上跳 1个单位长度,高度由 i 变为 i+1,且不能跳到其他的柱子上并且得不到该位置分数;
(2)向上跳 k个单位长度,高度由 i 变为i+k,且可以跳到任意柱子上并获得该位置上的分数;
(3)当以上两种操作都无法进行时,游戏结束并结算得分。
玩家初始时在单位长度 1处,且初始位置分数可以直接获得。请你求出最高能获得多少分。
输入描述
输出描述
示例
输入
5 3 2
1 4 3
2 2 5
4 3 4
7 2 7
1 2 1
输出
11
说明
从第2列开始,初始分数为4,向上移动一个单位长度(不获取分数)到达第2行第2列,再向上跳k=2个单位长度到达第4行第1列,或第4行第3列,总分数都是11
思路
很明显是动态规划,对于任意一个柱子的任意一个长度(长度1处除外),要么是从低一个单位长度的该柱子处跳上来的,要么就是从i-k长度处跳上来的(i>k),此时的状态方程怎么写呢?
首先设状态方程为dp[i][j],设从i-k长度处跳上来的为high_k[i]:
dp[i][j]=max(dp[i][j],dp[i-1][j]) //跳一个单位
high_k[i]=max(dp[i][j],high_k[i]) //跳一个单位的时候也要更新high_k[i]
当i>k:
dp[i][j]=max(high_k[i-k]+a[i][j],dp[i][j]) //跳k个单位,要加上此处的分数
最大分数什么时候找呢?跳不了的时候,那肯定是跳到了最高长度处咯^v^
具体见代码
代码
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e4;
int a[N][N],dp[N][N];
int high_k[N];
signed main(){
ios::sync_with_stdio(false),cin.tie(0); //记得关流,不然会超时
int n,m,k,ans=0;
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
cin>>a[i][j];
for(int j=1;j<=m;j++) //初始化一下dp
{
dp[1][j]=a[1][j]; //最开始的分数
high_k[1]=max(high_k[1],a[1][j]); //最开始肯定是选择最大的那个
}
for(int i=2;i<=n;i++)
for(int j=1;j<=m;j++)
{
dp[i][j]=max(dp[i][j],dp[i-1][j]); //跳一个单位长度
if(i>k)
{
dp[i][j]=max(high_k[i-k]+a[i][j],dp[i][j]); //k个单位长度
}
high_k[i]=max(dp[i][j],high_k[i]); //记得更新high_k[i]
}
for(int j=1;j<=m;j++)
ans=max(ans,dp[n][j]); //到最高的长度处肯定是跳不了了,所以最大分数在这里找
cout<<ans;
}