保留道路 Road

NOIP2016 模拟试题 复赛 提高组 day1

---

保留道路

(road.cpp/c/pas)

【问题描述】

很久很久以前有一个国家，这个国家有N个城市，城市由1,2,3,…,N标号，城市间有M条双向道路，每条道路都有两个属性g和s，两个城市间可能有多条道路，并且可能存在将某一城市与其自身连接起来的道路。后来由于战争的原因，国王不得不下令减小花费从而关闭一些道路，但是必须要保证任意两个城市相互可达。

道路花费的计算公式为wG*max{所有剩下道路的属性g}+wS*max{所有剩下道路的属性s}，其中wG和wS是给定的值。国王想要在满足连通性的前提下使这个花费最小，现在需要你计算出这个花费。

【输入格式】

输入文件名为road.in。

第一行包含两个正整数N和M。第二行包含两个正整数wG和wS。

后面的M行每行描述一条道路，包含四个正整数u,v,g,s，分别表示道路连接的两个城市以及道路的两个属性。

【输出格式】

输出文件名为road.out。

输出一个整数，表示最小花费。若无论如何不能满足连通性，输出-1。

【输入输出样例】

road.in

3 3
2 1
1 2 10 15
1 2 4 20
1 3 5 1

road.out

30

【数据规模与约定】

对于10%的数据，N≤10，M≤20；对于30%的数据，N≤100，M≤1000；对于50%的数据，N≤200，M≤5000；

对于100%的数据，N≤400，M≤50000，wG,wS,g,s≤1000000000。

---

代码系转载

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

typedef long long ll;
const int MAXN=4e2,MAXM=5e4;
const ll INF=1e17;
int N,M,wS,wG;
int cnt;
int fa[MAXN+1];
ll ans=INF;
struct node
{
    int x,y;bool vis;
    ll g,s;
} e[MAXM+1],tree[MAXM+1],tmp[MAXM+1];
bool cmp(const node & a,const node & b)
{
    if(a.g==b.g) return a.s<b.s;
    return a.g<b.g;
}
int find(int x)
{
    if(fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
void small(ll maxg)
{
    int i;
    for(i=1;i<=cnt;i++) tmp[i]=tree[i],tmp[i].vis=false;
    ll maxs=0,tot=0;
    for(i=1;i<=N;i++) fa[i]=i;//初始化并查集
    for(i=1;i<=cnt;i++)
    {
        int a=find(tree[i].x),b=find(tree[i].y);
        if(a!=b)
        {
            maxs=max(maxs,tree[i].s);
            fa[a]=b;
            tot++;
            tmp[i].vis=true;
        }
        if(tot==N-1)
        {
            int p=0;
            for(i=1;i<=cnt;i++)
                if(tmp[i].vis)
                    tree[++p]=tmp[i];
            cnt=p;
            ans=min(ans,maxg+maxs);
            break;
        }
    }
}
int main()
{
    freopen("road.in","r",stdin);
    freopen("road.out","w",stdout); 
    int i,j;
    cin>>N>>M>>wG>>wS;
    for(i=1;i<=M;i++)
    {
        scanf("%d%d%I64d%I64d",&e[i].x,&e[i].y,&e[i].g,&e[i].s);
        e[i].g*=(ll)wG,e[i].s*=(ll)wS;
    }
    sort(e+1,e+M+1,cmp);
    for(i=1;i<=M;i++)//遍历每条边 
    {
        if(e[i].g+e[i].s>ans) continue;//剪枝，
        //若现在的这条边的花费比之前算的ans还大就跳过 
        int pos=cnt+1;
        for(j=1;j<=cnt;j++)
            if(tree[j].s>e[i].s)
            {
                pos=j;
                break;
            }
        if(pos==cnt+1)
            tree[++cnt]=e[i];
        else
        {
            for(j=++cnt;j>=pos+1;j--)
                tree[j]=tree[j-1];
            tree[pos]=e[i];
        }
        small(e[i].g);
    }
    if(ans==INF) cout<<-1<<endl;
    else cout<<ans<<endl; 
    return 0;
}