必选边最小联络代价

【题目描述】
N个点，M条边，其中K条边必选，求所有点点之间都有可达的最小代价。数据保证有解。
【输入格式】
第一行两个整数n，m。
第二行到m+1行，每行四个非负整数，p,u,v,w。
当p=1时，表示必选边；
当p=2时，表示可选边；u,v,w一条无向边端点为u和v，权值为w。
【输出格式】
最小费用
【样例输入】
5 6
1 1 2 1
1 2 3 1
1 3 4 1
1 4 1 1
2 2 5 10
2 2 5 5
【样例输出】
9
【数据范围】
对于30%的数据，n<=10 m<=100
对于50%的数据, n<=200 m<=1000
对于100%的数据，n<=2000 m<=10000
【分析】
特殊的最小生成树，优先把所有必选的边添加进来，然后添加非必选的即可。

#include<bits/stdc++.h>
using namespace std;
const int N=2005;
const int M=10005;
int f[N];
struct Edge{
    int p,u,v,w;
}a[M];
int cmp(Edge x,Edge y){
    if (x.p==y.p) return x.w<y.w; else return x.p<y.p;
}
int get(int x){
    if (f[x]==x) return x;
    f[x]=get(f[x]);
    return f[x];
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++) scanf("%d%d%d%d",&a[i].p,&a[i].u,&a[i].v,&a[i].w);
    sort(a+1,a+1+m,cmp);
    int ans=0;
    for (int i=1;i<=m;i++) ans+=a[i].w;
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=m;i++){
        int root1=get(a[i].u),root2=get(a[i].v);
        if (a[i].p==1){f[root1]=root2;continue;}//直接添加
        if (root1==root2) ans-=a[i].w; else f[root1]=root2;
    }
    printf("%d\n",ans);
}