hdu 5807 Keep In Touch

思路：普通的状态转移复杂度是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]);
		}
	}
}