Sample Input
6
5 1
4 1
6 1
7 2
7 2
8 3
0
Sample Output
4
这是一道比较基础的dp练手题,规定arr[i][j]为第i秒j位置掉落的馅饼,dp[i][j]为第i秒时在j位置最多能够得到的馅饼数。
如果正向推导的话,每个状态应为此状态掉落的馅饼加上能够到达此状态的三个或两个状态中馅饼数最多的那个状态数。此状态转移方程如下:
d p [ i ] [ j ] = { a r r [ i ] [ j ] + m a x ( d p [ i − 1 ] [ j − 1 ] , d p [ i − 1 ] [ j ] , d p [ i − 1 ] [ j + 1 ] ) (j!=0&&j!=10) a r r [ i ] [ j ] + m a x ( d p [ i − 1 ] [ j ] , d p [ i − 1 ] [ j + 1 ] ) (j==0) a r r [ i ] [ j ] + m a x ( d p [ i − 1 ] [ j − 1 ] , d p [ i − 1 ] [ j ] ) (j==10) dp[i][j]= \begin{cases} arr[i][j]+max(dp[i-1][j-1],dp[i-1][j],dp[i-1][j+1])& \text{(j!=0\&\&j!=10)}\\ arr[i][j]+max(dp[i-1][j],dp[i-1][j+1])& \text{(j==0)}\\ arr[i][j]+max(dp[i-1][j-1],dp[i-1][j])& \text{(j==10)} \end{cases} dp[i][j]=⎩⎪⎨⎪⎧arr[i][j]+max(dp[i−1][j−1],dp[i−1][j],dp[i−1][j+1])arr[i][j]+max(dp[i−1][j],dp[i−1][j+1])arr[i][j]+max(dp[i−1][j−1],dp[i−1][j])(j!=0&&j!=10)(j==0)(j==10)
下面的maxtime记录的是最大的时间,如果把i<=maxtime改成i<=100000,提交代码也能够返回AC。
i<=maxtime的耗时是62ms,i<=100000的耗时是93ms。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=1e5+5;
int arr[N][15];//第几秒某个位置多少个馅饼
int dp[N][15];
int ans;
int main(void)
{
int n,x,t,maxtime;
while(scanf("%d",&n)!=EOF&&n)
{
maxtime=0;
memset(arr,0,sizeof arr);
memset(dp,0,sizeof dp);
for(int i=0;i<n;i++)
{
scanf("%d%d",&x,&t);
arr[t][x]++;
if(maxtime<t)
maxtime=t;
}
for(int j=0;j<=10;j++)//初始化
dp[0][j]=0;
for(int i=1;i<=maxtime;i++)//时间
for(int j=0;j<=10;j++)//位置
{
if(j>=5-i&&j<=5+i)
{
if(j!=0&&j!=10)
dp[i][j]=arr[i][j]+max(max(dp[i-1][j-1],dp[i-1][j]),dp[i-1][j+1]);
else if(j==0)
dp[i][j]=arr[i][j]+max(dp[i-1][j],dp[i-1][j+1]);
else
dp[i][j]=arr[i][j]+max(dp[i-1][j-1],dp[i-1][j]);
}
}
ans=0;
for(int j=0;j<=10;j++)
ans=max(ans,dp[maxtime][j]);
printf("%d\n",ans);
}
return 0;
}
这道题还可以用逆向推导
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=1e5+5;
int arr[N][15];//第几秒某个位置多少个馅饼
int dp[N][15];
int ans;
int main(void)
{
int n,x,t,maxtime;
while(scanf("%d",&n)!=EOF&&n)
{
maxtime=0;
memset(arr,0,sizeof arr);
memset(dp,0,sizeof dp);
for(int i=0;i<n;i++)
{
scanf("%d%d",&x,&t);
arr[t][x]++;
if(maxtime<t)
maxtime=t;
}
for(int j=0;j<=10;j++)//初始化
dp[0][j]=0;
for(int i=maxtime;i>=0;i--)//时间
for(int j=0;j<=10;j++)//位置
{
if(j!=0&&j!=10)
dp[i][j]=arr[i][j]+max(max(dp[i+1][j-1],dp[i+1][j]),dp[i+1][j+1]);
else if(j==0)
dp[i][j]=arr[i][j]+max(dp[i+1][j],dp[i+1][j+1]);
else if(j==10)
dp[i][j]=arr[i][j]+max(dp[i+1][j-1],dp[i+1][j]);
}
printf("%d\n",dp[0][5]);
}
return 0;
}