JZOJ100042. 【NOIP2017提高A组模拟7.12】保留道路

Description

很久很久以前有一个国家，这个国家有N个城市，城市由1,2,3,…,N标号，城市间有M条双向道路，每条道路都有两个属性g和s，两个城市间可能有多条道路，并且可能存在将某一城市与其自身连接起来的道路。后来由于战争的原因，国王不得不下令减小花费从而关闭一些道路，但是必须要保证任意两个城市相互可达。
道路花费的计算公式为wG*max{所有剩下道路的属性g}+wS*max{所有剩下道路的属性s}，其中wG和wS是给定的值。国王想要在满足连通性的前提下使这个花费最小，现在需要你计算出这个花费。

##分析
对于每条边，它有两个权值，
如果只有一个权值，那么可以用最小生成树。
现在我们就想办法就它变为一个权值。

我们先枚举g的最大值，
然后以s跑一次最小生成树，就可以求出是图联通的最小的s。
然而这样的时间复杂度是$O(m^2)$是不可以接受的。

我们考虑最小生成树的边，
如果某一条边在一次最小生成树中没有被选中，
在下一次，我们新增了一些边后，这条边依然不会被选中，因为已经有边比它更优。

也就是说在每次最小生成树后我们只需要保留n-1条边，即你选中的边。
对于每新插入一条边，只需要扫一遍将它插入，再跑最小生成树就可以。
这样的时间复杂度就变为了$O(nm)$

code

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#define N  50003
#define ll long long
using namespace std;
struct arr
{
    int x,y;
    ll s,g;
}a[N],b[N];
int n,m,f[N],f1,f2,p,l,r,mid;
ll ans,wg,ws,w[N];
char ch;
void read(int &n)
{
    n=0;
    ch=getchar();
    while((ch<'0' || ch>'9') && ch!='-')ch=getchar();
    int w=1;
    if(ch=='-')w=-1,ch=getchar();
    while('0'<=ch && ch<='9')n=n*10+ch-'0',ch=getchar();
    n*=w;
}
void readl(ll &n)
{
    n=0;
    ch=getchar();
    while((ch<'0' || ch>'9') && ch!='-')ch=getchar();
    int w=1;
    if(ch=='-')w=-1,ch=getchar();
    while('0'<=ch && ch<='9')n=n*10+ch-'0',ch=getchar();
    n*=w;
}
bool cmp(arr a,arr b)
{
    return a.s<b.s;
}
bool cmp1(arr a,arr b)
{
    return a.g<b.g;
}
int get(int x)
{
    if(x!=f[x])f[x]=get(f[x]);
    return f[x];
}
int main()
{
    read(n);read(m);readl(wg);readl(ws);
    for(int i=1;i<=m;i++)
        read(a[i].x),read(a[i].y),readl(a[i].g),readl(a[i].s);
    sort(a+1,a+1+m,cmp);
    ans=9223372036854775807;

    for(int i=1;i<n;i++)
        b[i]=a[i];
    sort(b+1,b+n,cmp1);

    for(int k=n-1;k<=m;k++)
    {
        if(a[k].s*ws>=ans)break;
        l=1;r=n-1;
        while(l<r)
        {
            mid=(l+r)/2;
            if(b[mid].g<a[k].g)l=mid+1;
                else r=mid;
        }
        for(int i=n;i>l;i--)
            b[i]=b[i-1];
        b[l]=a[k];
        for(int i=1;i<=n;i++)
            f[i]=i;
        p=0;
        for(int i=1;i<=n;i++)
        {
            f1=get(b[i].x);
            f2=get(b[i].y);
            if(f1!=f2)
            {
                f[f1]=f2;
                p++;
                b[p]=b[i];
                if(p==n-1)
                {
                    ans=min(ans,a[k].s*ws+b[p].g*wg);
                    break;
                }
            }
        }
    }
    if(ans==9223372036854775807)printf("-1");else printf("%lld",ans);
}