题意

有一棵 \(n\) 个点的树和 \(m\) 个人,第 \(i\) 个人从 \(u_i\) 走到 \(v_i\)

现在要发宠物,要求一个人要么他自己发到宠物,要么他走的路径上的都有宠物。

求最小代价,并输出任意方案。

\(n,m \leq 20000\)
传送门

思路

对每个人和每条树边都建一个点。

源点向每个人连容量 \(1\) 的边,每条树边向汇点连容量 \(1\) 的边。每个人向他要走到的所有边连容量 \(+\infty\) 的边。

给人发就是割掉人与源的边,放边上就是割树边与汇的边,人与树边间的边不能割。当源汇不连通的时候,就是满足题意的,题目转化为最小割。

问题出在连边上,这是 \(nm\) 的,考虑如何优化连边。线段树?树上问题有点麻烦,想到倍增。

对于一段相邻的\(2^i\)个点,都建一个点与它们所有相连,其实又有点像线段树,然后就会对于每条链得到一棵类似线段树的东西?大区间连向小区间,看起来就是。

然后对每个人,将他的路径在 LCA 处分成两条路径,这两条路径分别向对应区间覆盖。
那么我们得到的新图有 \(n\log n+m\) 个点和 \(n\log n+4m\) 条边,可以通过。

这里还要输出方案。

我们知道,最小割中的边一定满流。(全局最大流,分成两个集合,其间的边一定是满流的,而也就是最小割)因此我们从源点开始 dfs,只走没满流的边,并标记被 dfs 到的点。则图被分成两部分,一部分被访问过,一部分没被访问过。其中间那些边就是一个最小割。

然后,如果一个人代表的点没被访问过,则说明他所属那条边被割了。如果一条树边代表的点被访问过,则说明它所属那条边被割了。记录标号输出

#include <bits/stdc++.h>
using std::queue;
const int W=14,N=20005*(W+1),M=N+N*4;
int to[M<<1],w[M<<1],Next[M<<1],edge,n,x,y,f[20005][W+1],idn[N],deep[N],last[N],b[20005][W+1];
int s,t,m,cnt,ans,tag[N],cur[N];
queue <int> q;
void add(int x,int y,int z){
    to[++edge]=y;
    Next[edge]=last[x];
    last[x]=edge;
    w[edge]=z;
}
void dfs(int x,int fa){
    f[x][0]=fa,deep[x]=deep[fa]+1;
    for (int i=last[x];i;i=Next[i])
        if (to[i]!=fa) {
            idn[to[i]]=(i+1)/2;
            dfs(to[i],x);
        }
}
void Add(int x,int y,int w){
    add(x,y,w);
    add(y,x,0);
}
int lca(int x,int y){
    if (deep[x]<deep[y]) std::swap(x,y);
    for (int i=W;i>=0;i--)
        if (deep[f[x][i]]>=deep[y]) x=f[x][i];
    if (x==y) return x;
    for (int i=W;i>=0;i--)
        if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
void addedge(int x,int y,int id){
    for (int i=W;i>=0;i--)
        if (deep[f[x][i]]>=deep[y]){
            Add(id,b[x][i],n+1);
            x=f[x][i];
        }
}
bool bfs(){
    for (int i=0;i<=t;i++) cur[i]=last[i],deep[i]=0;
    deep[s]=1;
    q.push(s);
    while (!q.empty()){
        int x=q.front();
        q.pop();
        for (int i=last[x];i;i=Next[i])
            if (w[i] && !deep[to[i]]){
                deep[to[i]]=deep[x]+1;
                q.push(to[i]);
            }
    }
    return (deep[t]);
}
int c(int x) { return x&1?x+1:x-1; }
int dfs(int x){
    if (x==t) return 1;
    for (int i=cur[x];i;i=Next[i]){
        cur[x]=i;
        int u=to[i];
        if (deep[u]>deep[x] && w[i]){
            int di=dfs(to[i]);
            if (di){
                w[i]--;
                w[c(i)]++;
                return 1;
            }
        }
    }
    return 0;
}
void DFS(int x){
    tag[x]=1;
    for (int i=last[x];i;i=Next[i])
        if (w[i] && !tag[to[i]]) DFS(to[i]);
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        add(x,y,0),add(y,x,0);
    }
    s=0;
    dfs(1,0);
    edge=0;memset(last,0,sizeof(last));
    cnt=n;
    for (int i=2;i<=n;i++) b[i][0]=i;
    for (int i=1;i<=W;i++)
        for (int j=1;j<=n;j++)
            if (f[f[j][i-1]][i-1]){
                f[j][i]=f[f[j][i-1]][i-1];
                Add(++cnt,b[j][i-1],n+1);
                Add(cnt,b[f[j][i-1]][i-1],n+1);
                b[j][i]=cnt;
            }
    for (int i=1;i<=m;i++){
        scanf("%d%d",&x,&y);
        Add(s,++cnt,1);
        idn[cnt]=i;
        int l=lca(x,y);
        addedge(x,l,cnt),addedge(y,l,cnt);
    }
    t=cnt+1;
    for (int i=2;i<=n;i++) Add(i,t,1);
    while (bfs())
        while (dfs(s)==1) ans++;
    printf("%d\n",ans);
    DFS(s);
    ans=0;
    for (int i=last[s];i;i=Next[i])
        if (tag[to[i]]!=tag[s]) ans++;
    printf("%d ",ans);
    for (int i=last[s];i;i=Next[i])
        if (tag[to[i]]!=tag[s]) printf("%d ",idn[to[i]]);
    puts("");
    ans=0;
    for (int i=last[t];i;i=Next[i])
    if (tag[to[i]]!=tag[t]) ans++;
    printf("%d ",ans);
    for (int i=last[t];i;i=Next[i])
        if (tag[to[i]]!=tag[t]) printf("%d ",idn[to[i]]);
    return 0;
} 

后记

追随神仙的脚步
抄袭于此Mrsrz

02-09 18:25