HDOJ 5807 Keep In Touch

题意

所有的城市都有一个无线电波当两个城市的差值大于K时不能连通，给出三个人的初始位置，三个人要同时走到下一个城市，求有多少种保持联通的状态能达到。

思路

最暴力的想法就是n^6的暴力枚举这三个人当前在哪个城市要去哪个城市，显然复杂度太高了可以想办法降维，dp[i][j][k][l]表示现在这三个人在什么位置接下来要走的是第l个人的方案然后走了一个人推走第二个人，走了两个人推走第三个人这样推一下。不过写起来还是有不少小细节需要注意的。

代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
using namespace std;
#define LL unsigned long long
#define Lowbit(x) ((x)&(-x))
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1|1
#define MP(a, b) make_pair(a, b)
const int INF = 0x3f3f3f3f;
const int MOD = 998244353;
const int maxn = 200000 + 7;
const double eps = 1e-8;
const double PI = acos(-1.0);
typedef pair<int, int> pii;

void add(LL &x, LL y) {x = (x % MOD + y % MOD) % MOD;}

int n, m, K, q;
int w[55], mp[55][55];
LL dp[55][55][55][5];

bool check(int x, int y, int z)
{
    if (abs(w[x] - w[y]) > K) return 0;
    if (abs(w[x] - w[z]) > K) return 0;
    if (abs(w[y] - w[z]) > K) return 0;
    return 1;
}

int main()
{
    //freopen("H:\\in.txt","r",stdin);
    //freopen("H:\\out.txt","w",stdout);
    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(mp, 0, sizeof(mp));
        memset(dp, 0, sizeof(dp));
        for (int i = 0; i < m; i++)
        {
            int x, y;
            scanf("%d%d", &x, &y);
            mp[x][y] = 1;
        }
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n; j++)
                for (int k = 1; k <= n; k++)
                    if (check(i, j, k)) dp[i][j][k][2] = 1;
        for (int i = n; i >= 1; i--)
            for (int j = n; j >= 1; j--)
                for (int k = n; k >= 1; k--)
                {
                    for (int v = 1; v <= n; v++)
                        if (mp[i][v]) add(dp[i][j][k][0], dp[v][j][k][2]);
                    for (int v = 1; v <= n; v++)
                        if (mp[j][v]) add(dp[i][j][k][1], dp[i][v][k][0]);
                    if (dp[i][j][k][2] == 0) continue;            //注意
                    for (int v = 1; v <= n; v++)
                        if (mp[k][v]) add(dp[i][j][k][2], dp[i][j][v][1]);
                }
        while (q--)
        {
            int x, y, z;
            scanf("%d%d%d", &x, &y, &z);
            printf("%I64d\n", dp[x][y][z][2]);
        }
    }
    return 0;
}