题目描述
幸福国度可以用 N 个城镇(用 1 到 N 编号)构成的集合来描述,这些城镇 最开始由 M 条双向道路(用 1 到 M 编号)连接。城镇 1 是中央城镇。保证一个 人从城镇 1 出发,经过这些道路,可以到达其他的任何一个城市。这些道路都是 收费道路,道路 i 的使用者必须向道路的主人支付 ci分钱的费用。已知所有的这 些ci是互不相等的。最近有K条新道路建成,这些道路都属于亿万富豪Mr. Greedy。 Mr. Greedy 可以决定每条新道路的费用(费用可以相同),并且他必须在明天宣 布这些费用。
两周以后,幸福国度将举办一个盛况空前的嘉年华!大量的参与者将沿着这 些道路游行并前往中央城镇。共计 pj个参与者将从城镇 j 出发前往中央城镇。这 些人只会沿着一个选出的道路集合前行,并且这些选出的道路将在这件事的前一 天公布。根据一个古老的习俗,这些道路将由幸福国度中最有钱的人选出,也就 是 Mr. Greedy。同样根据这个习俗,Mr. Greedy 选出的这个道路集合必须使所有 选出道路的费用之和最小,并且仍要保证任何人可以从城镇 j 前往城镇 1(因此, 这些选出的道路来自将费用作为相应边边权的“最小生成树”)。如果有多个这样 的道路集合,Mr. Greedy 可以选其中的任何一个,只要满足费用和是最小的。
Mr. Greedy 很明确地知道,他从 K 条新道路中获得的收入不只是与费用有 关。一条道路的收入等于所有经过这条路的人的花费之和。更准确地讲,如果 p 个人经过道路 i,道路 i 产生的收入为 ci p 的积。注意 Mr. Greedy 只能从新道路 收取费用,因为原来的道路都不属于他。
Mr. Greedy 有一个阴谋。他计划通过操纵费用和道路的选择来最大化他的收 入。他希望指定每条新道路的费用(将在明天公布),并且选择嘉年华用的道路 (将在嘉年华的前一天公布),使得他在 K 条新道路的收入最大。注意 Mr. Greedy 仍然需要遵循选出花费之和最小的道路集合的习俗。
你是一个记者,你想揭露他的计划。为了做成这件事,你必须先写一个程序 来确定 Mr. Greedy 可以通过他的阴谋获取多少收入。
题解
我们需要让这k条边的代价最大。
看到k很小,可以想到枚举每条边的选择情况,但如果直接每次跑一遍最小生成树会使复杂度变得很高。
首先我们把这(k)条边先加进去,然后再对其他边跑最小生成树,那么此时加进去的边一定是必须在最后的生成树里的。
那么我们把这些边提前加进去,这样会形成(k+1)个联通块,我们把这些联通块看做点,那么点数的级别就变成了(O(K))的。
然后我们还发现有些边是没用的,我们先把联通块缩点,然后用(kruskal)尝试填加其他边,那么我们只会填加(k)条边,可以发现,此时还没有添加进去的边是没有用的,这些边就是可能会加进去的边,这样边数也是(O(K))的了。
对于新边的权值的设定,就是所有跨过它的边的权值最小值。
然后这道题就变成大模拟了。。
代码
#include<bits/stdc++.h>
#define N 100009
#define M 300009
#define K 22
#define inf 2e9
using namespace std;
typedef long long ll;
int f[N],n,m,k,l[N],r[N],dep[N],be[N],tot,num[N],tong[N],ma[K],mx[N],ji[K];
ll size[N],w[N],sum[N],ans;
bool vis[N];
int head[K],tott;
struct edge{
int n,to,l;
}e[K<<1];
inline void add(int u,int v,int l){
e[++tott].n=head[u];e[tott].to=v;head[u]=tott;e[tott].l=l;
}
inline ll rd(){
ll x=0;char c=getchar();bool f=0;
while(!isdigit(c)){if(c=='-')f=1;c=getchar();}
while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=getchar();}
return f?-x:x;
}
int find(int x){return f[x]=f[x]==x?x:find(f[x]);}
inline int zh(int x){return tong[be[x]];}
struct node{
int u,v,w;
inline bool operator <(const node &b)const{return w<b.w;}
}b[M];
ll dfs(int u,int fa){
size[u]=sum[u];
ll ans=0;
for(int i=head[u];i;i=e[i].n)if(e[i].to!=fa){
int v=e[i].to;
ans+=dfs(v,u);
size[u]+=size[v];
if(e[i].l)ans+=size[v]*mx[v];
}
return ans;
}
void dfs0(int u,int fa){
if(!fa)f[u]=u;
for(int i=head[u];i;i=e[i].n)if(e[i].to!=fa){
int v=e[i].to;f[v]=u;dep[v]=dep[u]+1;
dfs0(v,u);
}
}
int main(){
n=rd();m=rd();k=rd();
for(int i=1;i<=m;++i){
b[i].u=rd();b[i].v=rd();b[i].w=rd();
}
sort(b+1,b+m+1);
for(int i=1;i<=n;++i)f[i]=i;
for(int i=1;i<=k;++i){
l[i]=rd();r[i]=rd();
int xx=find(l[i]),yy=find(r[i]);
if(xx!=yy)f[xx]=yy;
}
for(int i=1;i<=n;++i)w[i]=rd();
for(int i=1;i<=m;++i){
int xx=find(b[i].u),yy=find(b[i].v);
if(xx!=yy)f[xx]=yy,num[++num[0]]=i,vis[i]=1;
}
for(int i=1;i<=n;++i)f[i]=i;
for(int i=1;i<=num[0];++i){
int xx=find(b[num[i]].u),yy=find(b[num[i]].v);
if(xx!=yy)f[xx]=yy;
}
for(int i=1;i<=n;++i){
be[i]=find(i);
if(!tong[be[i]])tong[be[i]]=++tot;
sum[tong[be[i]]]+=w[i];
}
for(int i=1;i<=m;++i){
int xx=find(b[i].u),yy=find(b[i].v);
if(xx!=yy){
f[xx]=yy;
ma[++ma[0]]=i;
}
}
for(int i=0;i<(1<<k);++i){
tott=0;ji[0]=0;
memset(head,0,sizeof(head));
for(int j=1;j<=tot;++j){
f[j]=j,dep[j]=0;mx[j]=inf;
}
for(int j=1;j<=k;++j)if(i&(1<<j-1)){
int xx=find(zh(l[j])),yy=find(zh(r[j]));
if(xx!=yy){
f[xx]=yy;
int x=zh(l[j]),y=zh(r[j]);
add(x,y,1);add(y,x,1);
}
}
for(int j=1;j<=ma[0];++j){
int xx=find(zh(b[ma[j]].u)),yy=find(zh(b[ma[j]].v));
if(xx!=yy){
f[xx]=yy;
int x=zh(b[ma[j]].u),y=zh(b[ma[j]].v);
add(x,y,0);add(y,x,0);
}
else{
ji[++ji[0]]=ma[j];
}
}
dfs0(zh(1),0);
for(int j=1;j<=ji[0];++j){
int xx=zh(b[ji[j]].u),yy=zh(b[ji[j]].v);
if(dep[xx]<dep[yy])swap(xx,yy);
while(dep[xx]>dep[yy])mx[xx]=min(mx[xx],b[ji[j]].w),xx=f[xx];
while(xx!=yy){
mx[xx]=min(mx[xx],b[ji[j]].w);
mx[yy]=min(mx[yy],b[ji[j]].w);
xx=f[xx];yy=f[yy];
}
}
ans=max(ans,dfs(zh(1),0));
}
printf("%lld
",ans);
return 0;
}