完美生成树——题解

题目大意：
在一个“无向有色图”中求一棵最小生成树，且要求树边必须包含至少两种颜色。
$1≤n点数,E边数≤100000,1≤color颜色,w边数≤1000000000$

第一眼看到数据范围简直令人绝望
不过，据大佬说这题只是NOIP普及组难度！！！关键是考试时居然写炸了！！！
但实际上很多时候这些巨大数值并无关紧要：hash，离散等，只要麻烦点预处理一下就OK了
当然，对于这题是不需要的
思考一下，显然先求他一趟最小生成树——
若此时颜色已经满足，那最好，直接出了答案
但若不满足呢？显然也是在这棵最小生成树上做一些变动来得到答案
那么关键就是如何变动了

——事实上，有且仅有一条原树边会被替换掉：
首先，显然，替换了一条树边后树边的总权值变大了（不然原来的就不叫最小生成树了）
那么假如我们再替换一条边，树边的总权值还会变大
所以说，有且仅有一条原树边会被替换掉

然后，这不就是变形的次小生成树吗？
加入一条边后，在这条边所形成的环中找一条最大的树边替换掉
快速找树上最大边——LCA另开个p数组记一下权值不就OK了吗？

于是：
Kruskal+建最小生成树
LCA+预处理p数组
枚举一条颜色不同的边“假装”加入，求此时次小生成树

PS：不得不提醒一下，最后刷次小生成树一定要另开变量记答案——求出的答案必大于等于原最小生成树！！！

#include<cstdio>
#include<algorithm>
#define LL long long
#define Max(x,y) max(x,y)
#define Min(x,y) (x)<(y)?(x):(y)
using namespace std;
const int maxn=(1e5)+5,Log=21;

int n,m,col=-1,fa[maxn];bool vis[maxn],flg=0;LL ans,val=((LL)1<<60);
int getfa(int x){return fa[x]==x?x:fa[x]=getfa(fa[x]);}

int tot,son[maxn<<1],nxt[maxn<<1],lnk[maxn],w[maxn<<1];
void add_e(int x,int y,int z){son[++tot]=y,w[tot]=z,nxt[tot]=lnk[x],lnk[x]=tot;}

int dep[maxn],up[maxn][Log],p[maxn][Log],pow[maxn];
void DFS(int x,int da,int va){
    dep[x]=dep[da]+1,up[x][0]=da,p[x][0]=va;
    for(int i=1;(1<<i)<=dep[x];i++) up[x][i]=up[up[x][i-1]][i-1],p[x][i]=Max(p[x][i-1],p[up[x][i-1]][i-1]);
    for(int j=lnk[x];j;j=nxt[j]) if(son[j]!=da) DFS(son[j],x,w[j]);
}
int LCA(int x,int y){
    int now=0;
    if(dep[x]<dep[y]) swap(x,y);
    while(dep[x]>dep[y]) now=Max(now,p[x][pow[dep[x]-dep[y]]-1]),x=up[x][pow[dep[x]-dep[y]]-1];
    if(x==y) return now;
    for(int k=pow[dep[x]];k>=0;k--)if(up[x][k]!=up[y][k])
        now=Max(Max(p[x][k],p[y][k]),now),x=up[x][k],y=up[y][k];
    return Max(Max(p[x][0],p[y][0]),now);
}

struct ff{
    int x,y,w,c;
    bool operator <(const ff b)const{return w<b.w;}
}E[maxn];
void work(){
    sort(E+1,E+1+m);
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1,t=0;i<=m&&t!=n-1;i++)if(getfa(E[i].x)!=getfa(E[i].y)){
        fa[fa[E[i].x]]=fa[E[i].y],ans+=E[i].w,++t,vis[i]=1;
        add_e(E[i].x,E[i].y,E[i].w),add_e(E[i].y,E[i].x,E[i].w);
        if(col<0) col=E[i].c;else if(col!=E[i].c) flg=1;
    }
}

char gt(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
int read(){
    int ret=0;char ch=gt();
    while(ch<'0'||ch>'9') ch=gt();
    while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=gt();
    return ret;
}

int main(){
    n=read(),m=read();
    for(int i=1;i<=m;i++) E[i]=(ff){read(),read(),read(),read()};
    work();
    if(flg){printf("%lld\n",ans);return 0;}
    DFS(1,0,0);
    for(int i=1;i<=n;i++) pow[i]=pow[i-1]+((1<<pow[i-1])==i);
    for(int i=1;i<=m;i++)if(!vis[i]&&col!=E[i].c) val=Min(val,ans-LCA(E[i].x,E[i].y)+E[i].w);
    printf("%lld\n",val);
    return 0;
}