题目链接:点这里!!!
题意:
给你n(n<=50)个城市,m(m<=n*(n-1)/2)条有向道路ui->vi(ui<vi)。有三个间谍去执行q(q<=125000)个任务,在每次任务中他们可能处于不同的城市,他们必须每时每刻在城市能用对讲机联系,i,j城市能联系当且仅当abs(wi-wj)<=k(k<=1e9)。每个时刻他们必须选一条道路去另一个城市,他们可以选择任意城市终止任务(三个人一起终止)。现在问针对每个任务有多少种不同的方案。
题解:
1、这道题的dp方程设的很巧妙,然后做起来就简单了很多。
2、我们设dp[0][i][j][k]为i,j,k在同一时刻的方案数为多少,dp[1][i][j][k]为i在上个时刻j,k在当前时刻的方案为多少,dp[2][i][j][k]为i,j在上个时刻,k在当前时刻的方案为多少。
3、我们可以得到他们的关系:dp[0]->dp[2],dp[1]->dp[0],dp[2]->dp[1],然后直接转移就ok了。注意判断下他们能不能联系的问题!
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<sstream>
#include<algorithm>
#include<vector>
#include<bitset>
#include<set>
#include<queue>
#include<stack>
#include<map>
#include<cstdlib>
#include<cmath>
#define LL long long
#define pb push_back
#define pa pair<int,int>
#define clr(a,b) memset(a,b,sizeof(a))
#define lson lr<<1,l,mid
#define rson lr<<1|1,mid+1,r
#define bug(x) printf("%d++++++++++++++++++++%d\n",x,x)
#define key_value ch[ch[root][1]][0]
#pragma comment(linker, "/STACK:102400000000,102400000000")
const LL MOD = 998244353;
const int N = 50+3;
const int maxn = 1e5+15;
const int letter = 130;
const int INF = 2e9;
const double pi=acos(-1.0);
const double eps=1e-6;
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int T,w[N],n,m,g,q;
int mp[N][N];
LL dp[3][N][N][N];
int main(){
scanf("%d",&T);
while(T--){
scanf("%d%d%d%d",&n,&m,&g,&q);
for(int i=1;i<=n;i++) scanf("%d",w+i);
clr(mp,0),clr(dp,0);
for(int i=0;i<m;i++){
int x,y;
scanf("%d%d",&x,&y);
mp[y][x]=1;
}
for(int i=n;i>=1;i--)
for(int j=n;j>=1;j--)
for(int k=n;k>=1;k--){
dp[0][i][j][k]=1,dp[1][i][j][k]=dp[2][i][j][k];
for(int t=1;t<=n;t++) if(mp[t][i]) dp[0][i][j][k]=(dp[0][i][j][k]+dp[1][t][j][k])%MOD;
for(int t=1;t<=n;t++) if(mp[t][j]) dp[1][i][j][k]=(dp[1][i][j][k]+dp[2][i][t][k])%MOD;
for(int t=1;t<=n;t++) if(mp[t][k]) dp[2][i][j][k]=(dp[2][i][j][k]+dp[0][i][j][t])%MOD;
if(abs(w[i]-w[k])>g) dp[0][i][j][k]=0;
if(abs(w[j]-w[k])>g) dp[0][i][j][k]=0;
if(abs(w[i]-w[j])>g) dp[0][i][j][k]=0;
}
while(q--){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
printf("%I64d\n",dp[0][x][y][z]);
}
}
return 0;
}