bzoj 3206: [Apio2013]道路费用 dfs+最小生成树+并查集

题意

有一个n个点m条边的无向图，边都有权值，每个点上有p[i]个人。现在又要新加入k条边，要求给这k条边赋上一个权值，然后在原图中选择任意一个最小生成树，求当所有人都要走到节点1时，k条边中每条边的权值乘上经过该边的人数的最大值。
n<=100000,m<=300000,k<=20

分析

一开始的想法是，2^k枚举每条边选或不选，然后强制选那些被选定的边。对于原来最小生成树上的边，若连接两个不同的集合，就连上去，否则就对路径上选定的边的权值进行更新。最后树形dp一遍就可以得出答案。
复杂度是$O(nk2^k)$。
考虑优化：
我们可以先强行选这k条边，然后做一遍最小生成树，这样最小生成树中的边无论如何都是比如会选的。这样我们就把图缩成了一个只有k+1个点k条边的图。
这时再用原图中的边做一次最小生成树。
然后就用上面的方法来做即可。
复杂度$O(mlogm+k^22^k)$
实现的时候细节较多。。。

分析

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=300005;
const int K=25;

int n,m,k,f[N],id[N],rt,tot;
LL p[N],sum,ans,val[K];
struct data{int u,v,w,op;}e[N],ex[K];
bool vis[K];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int find(int x)
{
    if (f[x]==x) return x;
    else return f[x]=find(f[x]);
}

bool cmp(data a,data b) {return a.w<b.w;}
bool cmpop(data a,data b) {return a.op>b.op||a.op==b.op&&a.w<b.w;}

void kruskal()
{
    sort(e+1,e+m+1,cmp);
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=k;i++) if (find(ex[i].u)!=find(ex[i].v)) f[find(ex[i].u)]=find(ex[i].v);
    for (int i=1;i<=m;i++)
    {
        int x=find(e[i].u),y=find(e[i].v);
        if (x!=y) f[x]=y,e[i].op=1;
    }
}

void build()
{
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=m;i++) if (e[i].op==1) p[find(e[i].v)]+=p[find(e[i].u)],f[find(e[i].u)]=find(e[i].v);
    for (int i=1;i<=n;i++) if (f[i]==i) id[i]=++tot,val[tot]=p[i];
    for (int i=1;i<=m;i++) e[i].u=id[find(e[i].u)],e[i].v=id[find(e[i].v)];
    for (int i=1;i<=k;i++) ex[i].u=id[find(ex[i].u)],ex[i].v=id[find(ex[i].v)];
    rt=id[find(1)];
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=m;i++)
    {
        int x=find(e[i].u),y=find(e[i].v);
        if (x!=y) e[i].op=2,f[x]=y;
    }
    sort(e+1,e+m+1,cmpop);
}

struct Tree_Dp
{
    int cnt,last[K];
    LL size[K];
    struct edge{int to,next,w,op;}e[K*2];

    void addedge(int u,int v,int op)
    {
        e[++cnt].to=v;e[cnt].op=op;e[cnt].w=0;e[cnt].next=last[u];last[u]=cnt;
        e[++cnt].to=u;e[cnt].op=op;e[cnt].w=0;e[cnt].next=last[v];last[v]=cnt;
    }

    bool dfs(int x,int fa,int y,int w)
    {
        if (x==y) return 1;
        for (int i=last[x];i;i=e[i].next)
        {
            if (e[i].to==fa) continue;
            if (dfs(e[i].to,x,y,w)) {if (!e[i].op&&!e[i].w) e[i].w=e[i^1].w=w;return 1;}
        }
        return 0;
    }

    void dp(int x,int fa)
    {
        size[x]=val[x];
        for (int i=last[x];i;i=e[i].next)
        {
            if (e[i].to==fa) continue;
            dp(e[i].to,x);
            size[x]+=size[e[i].to];
            if (!e[i].op) sum+=(LL)size[e[i].to]*e[i].w;
        }
    }
}tr;

void check()
{
    for (int i=1;i<=tot;i++) f[i]=i,tr.last[i]=0;tr.cnt=1;
    for (int i=1;i<=k;i++)
        if (vis[i])
        {
            if (find(ex[i].u)==find(ex[i].v)) return;
            tr.addedge(ex[i].u,ex[i].v,0);f[find(ex[i].u)]=find(ex[i].v);
        }
    for (int i=1;i<=tot-1;i++)
    {
        int x=find(e[i].u),y=find(e[i].v);
        if (x!=y) tr.addedge(e[i].u,e[i].v,1),f[x]=y;
        else tr.dfs(e[i].u,0,e[i].v,e[i].w);
    }
    sum=0;tr.dp(rt,0);ans=max(ans,sum);
}

void dfs(int x)
{
    if (x>k) {check();return;}
    dfs(x+1);
    vis[x]=1;dfs(x+1);vis[x]=0;
}

int main()
{
    n=read();m=read();k=read();
    for (int i=1;i<=m;i++) e[i].u=read(),e[i].v=read(),e[i].w=read();
    for (int i=1;i<=k;i++) ex[i].u=read(),ex[i].v=read();
    for (int i=1;i<=n;i++) p[i]=read();
    kruskal();
    build();
    dfs(1);
    printf("%lld",ans);
    return 0;
}