P3128 [USACO15DEC] LCA + 差分

最新推荐文章于 2022-10-27 16:05:16 发布

SHOHOKUKU

最新推荐文章于 2022-10-27 16:05:16 发布

阅读量130

点赞数

分类专栏：数据结构基本算法

本文链接：https://blog.csdn.net/neweryyy/article/details/109691620

版权

基本算法同时被 2 个专栏收录

252 篇文章 1 订阅

订阅专栏

数据结构

203 篇文章 1 订阅

订阅专栏

本文详细介绍了USACO竞赛中的P3128题目，该题目涉及树上的最大流问题。通过使用树上差分、倍增算法求解LCA（最近公共祖先），并动态维护路径覆盖次数。最终通过前缀和更新答案，得到覆盖次数最多的点。代码实现包括DFS遍历、LCA计算和路径贡献更新等关键步骤。

摘要由CSDN通过智能技术生成

题意

传送门 P3128 [USACO15DEC]Max Flow

题解

求树上被运输路线覆盖次数最多的点。考虑树上差分，倍增求 $L C A$ ，设路径的两个端点分别为 $u, v$ ，差分数组为 $d$ ，父节点数组为 $p a r$ ，则对于每一条路径
$d_u\leftarrow d_u+1$ $d_v\leftarrow d_v+1$ $d_{LCA}\leftarrow d_{LCA}-1$ $d_{par[LCA]}\leftarrow d_{par[LCA]}-1$ 最后回溯从叶子节点计算前缀和，更新答案即可。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define maxlg 16
#define maxn 50005
int N, K, res, lg[maxn], dif[maxn], par[maxlg][maxn], dep[maxn];
vector<int> G[maxn];

void dfs(int p, int v, int d)
{
    par[0][v] = p, dep[v] = d;
    for (int i = 1; i <= lg[d]; ++i)
        par[i][v] = par[i - 1][par[i - 1][v]];
    for (int i = 0; i < (int)G[v].size(); ++i)
    {
        int u = G[v][i];
        if (u != p)
            dfs(v, u, d + 1);
    }
}

int lca(int u, int v)
{
    if (dep[u] > dep[v])
        swap(u, v);
    while (dep[v] > dep[u])
        v = par[lg[dep[v] - dep[u]]][v];
    if (u == v)
        return v;
    for (int i = lg[dep[v]]; i >= 0; --i)
    {
        if (par[i][v] != par[i][u])
            v = par[i][v], u = par[i][u];
    }
    return par[0][v];
}

void sum(int p, int v)
{
    for (int i = 0; i < (int)G[v].size(); ++i)
    {
        int u = G[v][i];
        if (u != p)
        {
            sum(v, u);
            dif[v] += dif[u];
        }
    }
    res = max(res, dif[v]);
}

int main()
{
    scanf("%d%d", &N, &K);
    for (int i = 1; i < N; ++i)
    {
        int u, v;
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    lg[0] = -1;
    for (int i = 1; i <= N; ++i)
        lg[i] = lg[i - 1] + (1 << (lg[i - 1] + 1) == i);
    dfs(0, 1, 0);
    for (int i = 0; i < K; ++i)
    {
        int s, t;
        scanf("%d%d", &s, &t);
        int p = lca(s, t);
        ++dif[s], ++dif[t];
        --dif[p], --dif[par[0][p]];
    }
    sum(0, 1);
    printf("%d\n", res);
    return 0;
}