AtCoder Regular Contest 150 C - Path and Subsequence DP +最短路

C - Path and Subsequence DP +最短路

C - Path and Subsequence

赛时不会，赛后补题看了这位大佬的博客Konjac’s BlogA~D都有题解
本博客仅做个人的理解和记录

题意

给定 n 个点的无向连通图，每个点有一个点权 $ai$，同时给定一个长度为 $k$ 的序列 $b$，
询问是否任意从节点 $1$ 到节点 $n$ 的简单路径都满足如下性质：把经过的所有点的权值组成一个序列，$b$ 是这个序列的子序列。

解析

分析：因为询问的是对所有简单路径，那么可以考虑寻找对于b序列匹配得最少的，因为最后的结果是看匹配的最少的路径是否能完全匹配上$b$序列决定的，而不是匹配的更多的。
于是这些路径就有的 “长短” 的概念，可以进行djs最短路。
定义DP: $dp[i]$:从$1$~$i$的所有简单路径至少匹配$b$序列的前$dp[i]$位,即寻找匹配的最少的路径匹配多少
最后判断$dp[n]$是否等于k

难点：如何将匹配的多少抽象出来为最短路算法

代码

#include <queue>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10, inf = 1000000;
bool vis[N];
vector<int>g[N];
int n,m,k,a[N],b[N],dp[N];
typedef pair<int,int> PII;
priority_queue<PII, vector<PII>, greater<PII>>q;//小顶堆，PII = {dp[i], i} dp[i]更小的在堆顶
int main()
{
    cin >> n >> m >> k;
    for(int i = 1; i <= m; i ++){
        int u,v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }

    for(int i = 1; i <= n; i ++) cin >> a[i];
    for(int i = 1; i <= k; i ++) cin >> b[i];

    for(int i = 1; i <= n; i ++) dp[i] = inf;//初始化
    dp[1] = (a[1] == b[1]);

    q.push({dp[1], 1});
    while(!q.empty())//djs最短路
    {
        PII p = q.top();q.pop();
        int d = p.first, u = p.second;
        if(vis[u]) continue ;
        vis[u] = 1;

        for(int v : g[u])
        {
            //状态转移，当a[v]能与下一位b匹配上时，dp[v] = dp[u] + 1
            if(dp[v] > d + (a[v] == b[d + 1]))
            {
                dp[v] = d + (a[v] == b[d + 1]);
                q.push({dp[v], v});
            }
        }
    }
    cout << ((dp[n] == k) ? "Yes" : "No");
    return 0;
}