AtCoder Beginner Contest 265,266 部分dp题

题目大意：
一共有N次移动，每次可以从（x,y) 移动到：
（x+A，y+B）或 （x+C，y+D）或 （x+E，y+F）；
起始位置为（0，0），同时有m个点不能走，并给出m个点的坐标。
问一共有多少条不同路径（移动N次）。

初步想法为四维DP,即f [ t ] [ i ] [ j ] [ k ]表示前t步，移法一走了i次，移法二走了j次，移法三走了k次。
由于i+j+k=t, 因此可以去掉一维，题目要求的是方案数，因此状态转移方程为：

if(i) f[i][j][k]=(f[i-1][j][k]+f[i][j][k])%mod;
if(j) f[i][j][k]=(f[i][j-1][k]+f[i][j][k])%mod;
if(k) f[i][j][k]=(f[i][j][k-1]+f[i][j][k])%mod;

剩下的注意细节即可。
code:

#include<bits/stdc++.h>

using namespace std;
#define endl '\n'
#define ios std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define int long long
#define x first
#define y second
//head
typedef pair<int,int> pii;
const int N=3e2+10,mod=998244353;

map<pii,int> mp; 
int f[N][N][N];
void work()
{
	int n,m,a,b,c,d,e,ff;
	cin>>n>>m>>a>>b>>c>>d>>e>>ff;
	for(int i=1;i<=m;i++){
		int x,y;
		cin>>x>>y;
		mp[{x,y}]=1;
	}
	f[0][0][0]=1;
	for(int i=0;i<=n;i++){
		for(int j=0;j<=n;j++){
			for(int k=0;k<=n;k++){
				if(i+j+k>n) break;
				int x=i*a+j*c+k*e;
				int y=i*b+j*d+k*ff;
				if(mp[{x,y}]) f[i][j][k]=0;
				else  {
					if(i) f[i][j][k]=(f[i-1][j][k]+f[i][j][k])%mod;
					if(j) f[i][j][k]=(f[i][j-1][k]+f[i][j][k])%mod;
					if(k) f[i][j][k]=(f[i][j][k-1]+f[i][j][k])%mod;
				}
			}
		}
	}
	int ans=0;
	for(int i=0;i<=n;i++){
		for(int j=0;j<=n;j++){
			for(int k=0;k<=n;k++){
				if(i+j+k==n) ans=(ans+f[i][j][k])%mod;
			}
		}
	}
	cout<<ans<<endl;
}
signed main()
{
	ios;
	int t;
	t=1; //cin>>t;
	
	while(t--) work();
	
	return 0;
}

题目链接：
题目大意：
现在存在五个坑，下标从0到4，你现在在0的位置，每秒能移动一个距离，也可以不移动

有n个物品，会在时间t[i]的时候出现在x[i]的位置，且价值是a[i]

问你最多能获得的价值最大是多少

考虑dp, dp[i][j]表示时间为i，位置在j时所能得到的最大价值，则状态转移为：
dp[i][j]=dp[i-1][j]，即在第i秒不移动
dp[i][j]=dp[i-1][j-1],即从j-1移动到j
dp[i][j]=dp[i-1][j+1]即从j+1移动到j

注意转移的条件即可。

code:

#include<bits/stdc++.h>

using namespace std;
#define endl '\n'
#define ios std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define int long long
#define x first
#define y second
//head
typedef pair<int,int> pii;
const int N=1e5+10,mod=998244353;

int f[N][5];
int a[N];
int t[N],p[N];
void work()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		int tt,x,y;
		cin>>tt>>x>>y;
		a[tt]=y;
		p[tt]=x;
		//f[tt][x]=a[tt];
	}
	int ans=0;
	for(int i=1;i<5;i++) f[0][i]=-1e18; //初始化 
	
	for(int i=1;i<=1e5;i++){
		
		for(int j=0;j<5;j++){
			f[i][j]=f[i-1][j];
			if(j) f[i][j]=max(f[i][j],f[i-1][j-1]);
			if(j!=4) f[i][j]=max(f[i][j],f[i-1][j+1]); 
		}
		f[i][p[i]]+=a[i];
	}
	for(int i=0;i<5;i++){
		ans=max(ans,f[100000][i]);
	}
	cout<<ans<<endl;
}
signed main()
{
	int t;
	t=1; //cin>>t;
	
	while(t--) work();
	
	return 0;
}