APIO2013 道路费用

题目链接
大意是说，调整k条特殊边边的值，并在原图的基础上构建最小生成树。使得经过所有特殊边的值之和最大。

感谢dasxxx学长指出题目要素(加上新边，两点之间最多只有一条边)，请各位在理解下文解法时特别注意，如果没有这一要求是不能设计如下算法的。在此安利一下HGEEK首席程序员的博客。

大致解法：
先缩点。可以知道，对答案有贡献的只有k条新边。先将k条边进行最小生成树，再构建一棵完成的最小生成树。
除去那些新边会产生k+1个连通块，而这连通块中的边的连接方法对答案是没有贡献的，起到主要贡献的是连接这些连通块的新边。当然也会有k条原边可以替代它们。

先进行暴力枚举k条边的选或者不选，总共$2^k$种状态。用选定的新边将连通块相连，不足的部分用原边代替。那么在这棵树中主要是求到每个点的人数和所有新边的最大值。每点人数用树上前缀和即可；求新边最值需要借用kruskal的思想，即在树上从u到v的路径中的所有边权必须小于等于新加进的边，不然就可以用新边代替之，如此约束每条新边的值即可。

code:

#include<bits/stdc++.h>
typedef long long ll;
const int INF=1<<30;
using namespace std;
ll ans;
int n,m,K,top,cnt,st;
int fa[100005],fa2[100005],p[100005];
struct edgege{
    int u,v,w;
}e[300005],ne[25],q[300005];
struct data{
    int v,nxt;
}edge[50];
int ag[25],dep[100005],head[100005],mn[100005];
ll val[100005],sum[100005];
bool mark[300005];
void insert(int u,int v)
{
    edge[++cnt]=(data){v,head[u]};head[u]=cnt;
    edge[++cnt]=(data){u,head[v]};head[v]=cnt;
}
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
int find2(int x){return x==fa2[x]?x:fa2[x]=find2(fa2[x]);}
bool operator<(edgege a,edgege b){return a.w<b.w;}
void dp(int x)
{
    sum[x]=val[x];
    for(int i=head[x];i;i=edge[i].nxt)
        if(edge[i].v!=fa2[x])
        {
            dep[edge[i].v]=dep[x]+1;
            fa2[edge[i].v]=x;
            dp(edge[i].v);
            sum[x]+=sum[edge[i].v];
        }
}
void solve()
{
    cnt=0;
    for(int i=1;i<=K+1;i++)
    {
        int p=ag[i];
        head[p]=fa2[p]=0;
        fa[p]=p;mn[p]=INF;
    }
    for(int i=1;i<=K;i++)
        if(mark[i])
        {
            int x=find(ne[i].u),y=find(ne[i].v);
            if(x==y)return;
            fa[x]=y;
            insert(ne[i].u,ne[i].v);
        }//将新边加入MST 
    for(int i=1;i<=K;i++)
    {
        int x=find(q[i].u),y=find(q[i].v);
        if(x!=y)fa[x]=y,insert(q[i].u,q[i].v);
    }//构建完整的MST 
    dp(st);//树上前缀和(人数)
    for(int i=1;i<=K;i++)
    {
        int u=q[i].u,v=q[i].v;
        if(dep[u]>dep[v])swap(u,v);
        while(dep[v]>dep[u])mn[v]=min(mn[v],q[i].w),v=fa2[v];
        while(u!=v)
        {
            mn[v]=min(mn[v],q[i].w);
            mn[u]=min(mn[u],q[i].w);
            u=fa2[u];v=fa2[v];
        }
    }//退化版lca 
    ll res=0;
    for(int i=1;i<=K;i++)
        if(mark[i])
        {
            int u=ne[i].u,v=ne[i].v;
            if(dep[u]>dep[v])swap(u,v);
            res+=mn[v]*sum[v];
        }
    ans=max(res,ans);
}
void dfs(int x)
{//选边的状态枚举 
    if(x>K)
    {
        solve();return;
    }
    mark[x]=0;dfs(x+1);
    mark[x]=1;dfs(x+1);
}
int main()
{
    scanf("%d%d%d",&n,&m,&K);
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
    sort(e+1,e+m+1);
    for(int i=1;i<=K;i++)
        scanf("%d%d",&ne[i].u,&ne[i].v);
    for(int i=1;i<=n;i++)
        scanf("%d",&p[i]);
    for(int i=1;i<=n;i++)
        fa[i]=fa2[i]=i;
    for(int i=1;i<=K;i++)
        fa[find(ne[i].u)]=find(ne[i].v);//新边的缩点 
    for(int i=1;i<=m;i++)
    {
        int u=e[i].u,v=e[i].v;
        if(find(u)!=find(v))//一定会借助其他边连接的两点 
        {
            fa[find(u)]=fa[find(v)];
            fa2[find2(u)]=fa2[find2(v)];
        }
    }
    //这里find合并成了一棵树
    //find2是K+1块 
    st=find2(1);//初始1所在集合 
    for(int i=1;i<=n;i++)
    {
        val[find2(i)]+=p[i];//每块的人数 
        if(find2(i)==i)ag[++ag[0]]=i;//每个缩点(集合)的编号
    }
    for(int i=1;i<=K;i++)
        ne[i].u=find2(ne[i].u),ne[i].v=find2(ne[i].v); 
    for(int i=1;i<=m;i++)
        e[i].u=find2(e[i].u),e[i].v=find2(e[i].v);
    //重构边，负责连块 
    for(int i=1;i<=m;i++)
    {
        int p=find2(e[i].u),q=find2(e[i].v);
        if(p!=q)mark[i]=1,fa2[p]=q;
    }
    for(int i=1;i<=m;i++)
        if(mark[i])q[++top]=e[i];//取那些有用的边
    memset(mark,0,sizeof(mark));
    dfs(1);
    printf("%lld\n",ans);
    return 0;
}