思路:普通的状态转移复杂度是n^6,这里学习了别人的做法,分段dp, 复杂度降到n^4
dp[i][j][k][0]代表下一步走i
dp[i][j][k][1]代表下一步走j
dp[i][j][k][2]代表下一步走k
状态转移部分如下
if(dp[i][j][k][0])
for(int p=1;p<k;p++)
if(e[p][k])
add(dp[i][j][p][2],dp[i][j][k][0]);
if(dp[i][j][k][2])
for(int p=1;p<j;p++)
if(e[p][j])
add(dp[i][p][k][1],dp[i][j][k][2]);
if(dp[i][j][k][1])
for(int p=1;p<i;p++)
if(e[p][i])
add(dp[p][j][k][0],dp[i][j][k][1]);
下面是程序代码:
#include <iostream>
#include <cstring>
#include <stdio.h>
#include <cmath>
using namespace std;
int e[60][60],dp[60][60][60][4];
int w[60];
int n,m,k,q;
int mod=998244353;
void add(int &a,int b)
{
a=(a+b)%mod;
}
int ok(int x,int y)
{
if(abs(w[x]-w[y])>k)
return 0;
return 1;
}
int main()
{
//freopen("in.txt","r",stdin);
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d%d",&n,&m,&k,&q);
for(int i=1;i<=n;i++)
scanf("%d",&w[i]);
memset(e,0,sizeof(e));
int x,y,z;
for(int i=0;i<m;i++)
{
scanf("%d%d",&x,&y);
e[x][y]=1;
}
memset(dp,0,sizeof(dp));
for(int i=n;i>0;i--)
{
for(int j=n;j>0;j--)
{
for(int k=n;k>0;k--)
{
if(!ok(i,j) || !ok(i,k) || !ok(j,k))dp[i][j][k][0]=0;
else add(dp[i][j][k][0],1);
if(dp[i][j][k][0])
for(int p=1;p<k;p++)
if(e[p][k])
add(dp[i][j][p][2],dp[i][j][k][0]);
if(dp[i][j][k][2])
for(int p=1;p<j;p++)
if(e[p][j])
add(dp[i][p][k][1],dp[i][j][k][2]);
if(dp[i][j][k][1])
for(int p=1;p<i;p++)
if(e[p][i])
add(dp[p][j][k][0],dp[i][j][k][1]);
}
}
}
while(q--)
{
scanf("%d%d%d",&x,&y,&z);
printf("%d\n",dp[x][y][z][0]);
}
}
}