CF1675 F.Vlad and Unfinished Business(dfs + 思维)

大致题意 : 给定起点x, 终点y, k个任务点a1...ak, 以及一棵n个点, n-1条边的数, 问经过k个任务点, x->y的最短路

思路 : 对 x->y(经过k)的路径(即答案路径) 中所有点进行染色(标记), 表示这些点一定会经过, 答案即为2*(cnt-1)-d[y];

因为除了x->y的最短路这条路径的边外, 其他的边最少会经过两次(x到任务点后还会经过此边返回去找其他任务点, 注意:只针对不在x->y最短路上的边), 由此得出需经过的边数为2*(cnt-1)(cnt-1即,点数-1 就是边数) -d[y](d[y]为x->y的最短路, 因为最短路只用走一遍所以要减去一次)

代码及其注释 :

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
#include <iterator>
#include <cmath>
#include <algorithm>
#include <sstream>
#include <map>
#include <stack>
#include <set>
#include <queue>
#include <iomanip>
using namespace std;

stringstream ss;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 2e5+10, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int T, d[N], ans;
bool st[N], vis[N];
vector<int> v[N];

// 对x->y(经过k)的路径(即答案路径)中所有点进行染色(标记), 即这些点一定会经过, 答案即为2*(cnt-1)-d[y]
// 因为除了x->y的最短路这条路径的边外
// 其他的边最少会经过两次(x到任务点后还会经过此边返回去找其他任务点, 注意:只针对不是x->y最短路上的边)
// 由此得出需经过的边数为2*(cnt-1)(cnt-1即点数-1就是边数)-d[y](d[y]为x->y的最短路, 因为最短路只用走一遍)
bool dfs(int u,int fu)
{
    // 记录路径长度, 也可以再在dfs上开一个deep即dfs(u,fu,deep)
    if(u != fu) d[u] = d[fu] + 1;
    // 本就是要到的点直接标记
    vis[u] = st[u];
    for(int i = 0; i<v[u].size(); i++)
    {
        int j = v[u][i];
        // dfs判断这条边是否合法, 即这条边是否必经, 是就给点打上标记
        if(j != fu && dfs(j, u)) vis[u] = 1;
    }
    return vis[u];
}
int main()
{
    scanf("%d", &T);
    while(T -- )
    {
        int n,k;
        scanf("%d%d", &n, &k);
        int x,y;
        scanf("%d%d", &x, &y);
        for(int i = 1; i<=n; i++)
        {
            vis[i] = d[i] = st[i] = 0;
            v[i].clear();
        }
        for(int i = 1; i<=k; i++)
        {
            int a;
            cin>>a;
            st[a] = 1;
        }
        for(int i = 1; i<=n-1; i++)
        {
            int a,b;
            cin>>a>>b;
            v[a].push_back(b);
            v[b].push_back(a);
        }
        st[y] = 1;
        int cnt = 0;
        dfs(x, x);
        // cnt记录点数
        for(int i = 1; i<=n; i++)
        {
            cnt += vis[i];
        }
        cout<<2*(cnt-1)-d[y]<<"\n";
    }
}