F缆车（lca + 树形dp）

链接：

F-缆车

题意：

给定一棵树，树根是1，树的边都是单向边。再给定树上的m个点和点K，现在可以从K连一条边到某个点，求连接完后K到这m个点的最小路径和。

做法：

求一下k点与这m个点的lca，如果这些点的lca不是k，那k必然连向那m个点的lca，k可能在这m个点之中，也可能不在，这个在细节上注意下就行。
如果lca是k，那就枚举k向k的所有儿子连边，看哪种情况路径和最小就行了。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e6 + 5, mod = 1e9 + 7;
vector<int> v[N];
int d[N], f[N][30];
int p[N];
int t;
void bfs()
{
    queue<int> q;
    q.push(1);
    d[1] = 1;
    while (!q.empty())
    {
        int x = q.front();
        q.pop();
        for (auto i : v[x])
        {
            int y = i;
            if (d[y]) continue;
            d[y] = d[x] + 1;
            f[y][0] = x;
            for (int j = 1; j <= t; j++)
            {
                f[y][j] = f[f[y][j - 1]][j - 1];
            }
            q.push(y);
        }
    }
}
int LCA(int x, int y)
{
    if (d[x] > d[y])
        swap(x, y);
    for (int i = t; i >= 0; i--)
    {
        if (d[f[y][i]] >= d[x]) y = f[y][i];
    }
    if (x == y) return x;
    for (int i = t; i >= 0; i--)
    {
        if (f[x][i] != f[y][i])
        {
            x = f[x][i];
            y = f[y][i];
        }
    }
    return f[x][0];
}
int ws(int n)
{
    int ret = 0;
    while (n)
    {
        n >>= 1;
        ret++;
    }
    return ret;
}
int ans, sum, k;
int siz[N];
void dfs1(int x)
{
    for (auto i : v[x])
    {
        dfs1(i);
        siz[x] += siz[i];
    }
    if (d[x] > d[k]) ans = min(ans, sum - siz[x] * (d[x] - d[k] - 1));
}
void solve()
{
    int n;
    cin >> n;
    t = ws(n);
    for (int i = 1; i <= n - 1; i++)
    {
        int x, y;
        scanf("%lld%lld", &x, &y);
        v[x].push_back(y);
    }
    bfs();
    int m;
    cin >> m >> k;
    for (int i = 1; i <= m; i++)
    {
        scanf("%lld", &p[i]);
        siz[p[i]] = 1;
    }

    if (m == 1)
    {
        if (k == p[1]) cout << 0;
        else cout << 1;
        return;
    }
    if (p[1] == k) swap(p[1], p[m]);
    int llccaa = p[1];
    int lp = p[1];
    for (int i = 1; i <= m; i++)
    {
        llccaa = LCA(llccaa, p[i]);
        if (k != p[i]) lp = LCA(lp, p[i]);
    }
    llccaa = LCA(llccaa, k);
    ans = 0;
    if (llccaa != k)
    {
        for (int i = 1; i <= m; i++)
        {
            if (k != LCA(k, p[i]))
            {
                ans += d[p[i]] - d[lp] + 1;
            }
            else ans += d[p[i]] - d[k];
        }
    }
    else
    {
        for (int i = 1; i <= m; i++)
        {
            ans += d[p[i]] - d[k];
        }
        sum = ans;
        dfs1(k);
    }
    cout << ans;
}
signed main()
{
    int tt = 1;
    // cin >> tt;
    while (tt--) solve();
    return 0;
}