P3128 [USACO15DEC]最大フローP(ツリーチェーン分割+ラインセグメントツリーメンテナンス最大)

https://www.luogu.com.cn/problem/P3128


 

FJは、牛舎のN(2≤N≤50,000)コンパートメントの間にN-1パイプを設置しました。コンパートメント番号は1からNです。すべてのコンパートメントはパイプで接続されています。

FJには牛乳を輸送するためのK(1≤K≤100,000)ルートがあり、i番目のルートはコンパートメントsiからコンパートメントtiに輸送されます。輸送ルートは、輸送圧力の単位を2つの端点のコンパートメントと中央のすべてのコンパートメントにもたらします。最もストレスのかかるコンパートメントの圧力を計算する必要があります。


アイデア:ツリーチェーンが分割された後、ラインセグメントツリーの操作とメンテナンスの間隔が最適な値になります。+ 1操作が実行されるたびに、タグ付けされたサブツリーmaxval + 1、親maxvalが2つのサブツリーの最大maxvalから転送されます。最後に、tree [1] .maxvalを直接確認します。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=5e4+100;
typedef long long LL;
struct Tree{
    LL l,r,maxval,tag;
}tree[maxn*4];
LL a[maxn];
///线段树部分
void push_up(LL p){
    tree[p].maxval=max(tree[p*2].maxval,tree[p*2+1].maxval);
}
void addtag(LL p,LL d){
    tree[p].tag+=d;
    tree[p].maxval+=d;
}
void push_down(LL p){
    if(tree[p].tag!=0){
        addtag(p*2,tree[p].tag);
        addtag(p*2+1,tree[p].tag);
        tree[p].tag=0;
    }
}
void build(LL p,LL l,LL r){
    tree[p].l=l;tree[p].r=r;tree[p].tag=0;tree[p].maxval=0;
    if(l==r){return;}
    LL mid=(l+r)>>1;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    push_up(p);
}
void modify(LL p,LL l,LL r,LL d){
    if(l<=tree[p].l&&r>=tree[p].r){
        addtag(p,d);
        return;
    }
    push_down(p);
    LL mid=(tree[p].l+tree[p].r)>>1;
    if(l<=mid) modify(p*2,l,r,d);
    if(r>mid) modify(p*2+1,l,r,d);
    push_up(p);
}
LL query(LL p,LL l,LL r){
    return tree[1].maxval;
}
///树剖部分
LL son[maxn],siz[maxn],dep[maxn],fa[maxn];
LL top[maxn];
LL id[maxn],tot=0;
LL new_w[maxn];
vector<LL>g[maxn];
void predfs(LL u,LL father){
    siz[u]=1;dep[u]=dep[father]+1;
    fa[u]=father;
    for(LL i=0;i<g[u].size();i++){
        LL v=g[u][i];
        if(v==father) continue;
        predfs(v,u);
        siz[u]+=siz[v];
        if(siz[v]>siz[son[u]]){
            son[u]=v;
        }
    }
}
void dfs(LL u,LL topx){
    id[u]=++tot;
    top[u]=topx;
    new_w[tot]=a[u];
    if(!son[u]) return;///叶子节点访问完了
    dfs(son[u],topx);///先处理重儿子
    for(LL i=0;i<g[u].size();i++){
        LL v=g[u][i];
        if(v==fa[u]||v==son[u]) continue;
        dfs(v,v);
    }
}
void modify_path(LL u,LL v,LL d)
{
    while(top[u]!=top[v]){
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        modify(1,id[top[u]],id[u],1);
        u=fa[top[u]];
    }
    if(dep[u]>dep[v]) swap(u,v);
    modify(1,id[u],id[v],1);
}
LL query_path(LL u,LL v){
    return tree[1].maxval;
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL n,m;cin>>n>>m;
  for(LL i=1;i<n;i++){
    LL u,v;cin>>u>>v;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  predfs(1,0);
  dfs(1,0);
  build(1,1,n);
  while(m--){
    LL u,v;cin>>u>>v;
    modify_path(u,v,1);
  }
  cout<<query_path(1,n)<<endl;
return 0;
}

 

おすすめ

転載: blog.csdn.net/zstuyyyyccccbbbb/article/details/109864107
おすすめ