分步DP--HDU5807Keep In Touch

本文主要是介绍分步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;
}

这篇关于分步DP--HDU5807Keep In Touch的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---