HDU5807 Keep In Touch DP

// HDU5807 Keep In Touch   DP
// 思路：直接暴力是O(n^6).所以要优化一下
// dp[i][j][k][0]：当前点i j k的方案数
// dp[i][j][k][1]：j在当前时刻，i k还在上次
// dp[i][j][k][2]:j k在当前时刻，i还在上次
// 那么就可以转移了 本题u<v 所以转移的时候从大到小 从后往前

#include <bits/stdc++.h>
using namespace std;
#define clc(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int N = 300010; 
const int MOD = 998244353;
#define LL long long
double const pi = acos(-1);
void fre() {freopen("in.txt","r",stdin);}

int n,m,c,q;
int w[55];
int g[55][55];
int dp[55][55][55][3];

int check(int a,int b,int c){
   return max(abs(a-b),max(abs(b-c),abs(a-c)));  
}
void DP(){
    for(int i=n;i>=1;i--){
        for(int j=n;j>=1;j--){
            for(int k=n;k>=1;k--){
               dp[i][j][k][0]=1,dp[i][j][k][1]=dp[i][j][k][2]=0;
               for(int h=i+1;h<=n;h++){
                  if(g[h][i]) dp[i][j][k][0]=(dp[i][j][k][0]+dp[h][j][k][2])%MOD;
               }
               for(int h=j+1;h<=n;h++){
                  if(g[h][j]) dp[i][j][k][1]=(dp[i][h][k][0]+dp[i][j][k][1])%MOD;
               }
               for(int h=k+1;h<=n;h++){
                  if(g[h][k]) dp[i][j][k][2]=(dp[i][j][h][1]+dp[i][j][k][2])%MOD;
               }
               if(check(w[i],w[j],w[k])>c) dp[i][j][k][0]=0;
            }
        }
    }
}

int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d%d",&n,&m,&c,&q);
        for(int i=1;i<=n;i++) scanf("%d",&w[i]);
        clc(g,0);
        clc(dp,0);
        for(int i=1;i<=m;i++) {
            int u,v;
            scanf("%d%d",&u,&v);
            g[v][u]=1;
        }
        DP();
        while(q--){
            int u,v,z;
            scanf("%d%d%d",&u,&v,&z);
            printf("%d
",dp[u][v][z][0]);
        }
    }
    return 0;
}