分步DP--HDU5807Keep In Touch

传送门
暴力地想设$f[i][j][k]$表示三个人从$i,j,k$出发的方案数，复杂度$n^6$

可以用加维优化的思想
设$f[i][j][k][l]$表示当前三个人走到$i,j,k$，该$l$走的方案数
这样状态数是$n^3$，转移$O(n)$，总复杂度$O(n^4)$
因为$u<v$所以可以倒着$dp$，最后$f[i][j][k][0]$就是从$i,j,k$出发的方案数

然后注意些小细节就好了，比如当前方案不可行的时候要设成$0$，这样后面就相当于没有加

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#define N 55
using namespace std;
int T,n,m,K,q,f[N][N][N][3],val[N];
const int mod=998244353;
vector<int> vec[N];

inline int rd(){
	int x=0,f=1;char c=' ';
	while(c<'0' || c>'9') f=c=='-'?-1:1,c=getchar();
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*f;
}

inline bool check(int x,int y,int z){
	return abs(val[x]-val[y])<=K&&abs(val[y]-val[z])<=K&&abs(val[x]-val[z])<=K;
}

inline void DP(){
	for(int i=n;i;i--)
		for(int j=n;j;j--)
			for(int k=n;k;k--){
				if(check(i,j,k)) (++f[i][j][k][0])%=mod;
				else f[i][j][k][0]=0;//一定要加这句qwq 
				for(int t=0;t<vec[i].size();t++){
					int x=vec[i][t];
					(f[x][j][k][1]+=f[i][j][k][0])%=mod;
				}
				for(int t=0;t<vec[j].size();t++){
					int x=vec[j][t];
					(f[i][x][k][2]+=f[i][j][k][1])%=mod;
				}
				for(int t=0;t<vec[k].size();t++){
					int x=vec[k][t];
					(f[i][j][x][0]+=f[i][j][k][2])%=mod;
				}
				if(!check(i,j,k)) f[i][j][k][0]=0; 
			}
}

int main(){
	T=rd();
	while(T--){
		n=rd(); m=rd(); K=rd(); q=rd();
		for(int i=1;i<=n;i++) val[i]=rd(),vec[i].clear();
		for(int i=1;i<=m;i++){
			int x=rd(),y=rd();
			vec[y].push_back(x);//反边 
		}
		memset(f,0,sizeof f);
		DP();
		while(q--){
			int x=rd(),y=rd(),z=rd();
			printf("%d\n",f[x][y][z][0]);
		}
	}
	return 0;
}