【UOJ#236】[IOI2016]railroad(欧拉回路,最小生成树)

【UOJ#236】[IOI2016]railroad(欧拉回路,最小生成树)

题面

UOJ

题解

把速度看成点,给定的路段看成边,那么现在就有了若干边,然后现在要补上若干边,以及一条\([inf,\)使得原图存在欧拉回路,那么就变成了求从大往小连边的边长的最小值。
而欧拉回路每个点被来回覆盖的次数左右一定是一样的,假设向右-向左覆盖的次数为\(g_i\),那么如果\(g_i>0\),花费\(1\)的代价向\(i-1\)连边,如果\(g_i>0\),那么则可以不花费代价连边\(i\rightarrow i+1\)
看起来这样子得到了一个解,实际上欧拉回路还需要满足连通性,再求一遍\(MST\)把图连通就行了。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include"railroad.h"
using namespace std;
#define MAX 400200
#define ll long long
int S[MAX<<2],top,cnt,c[MAX];
int f[MAX];int getf(int x){return x==f[x]?x:f[x]=getf(f[x]);}
struct Line{int u,v,w;}e[MAX];
bool operator<(Line a,Line b){return a.w<b.w;}
long long plan_roller_coaster(vector<int> s,vector<int> t)
{
    int n=s.size();ll ans=0;
    for(int i=0;i<n;++i)S[++top]=s[i],S[++top]=t[i];
    S[++top]=-1e9-100;S[++top]=1e9+100;
    sort(&S[1],&S[top+1]);top=unique(&S[1],&S[top+1])-S-1;
    for(int &i:s)i=lower_bound(&S[1],&S[top+1],i)-S;
    for(int &i:t)i=lower_bound(&S[1],&S[top+1],i)-S;
    c[2]=-1;c[top]=1;for(int i=1;i<=top;++i)f[i]=i;
    for(int i=0;i<n;++i)c[s[i]+1]++,c[t[i]+1]--;
    for(int i=0;i<n;++i)f[getf(s[i])]=getf(t[i]);
    for(int i=1;i<=top;++i)c[i]+=c[i-1];
    for(int i=2;i<=top;++i)
    {
        if(c[i]==0)continue;
        if(c[i]>0)ans+=1ll*c[i]*(S[i]-S[i-1]);
        f[getf(i)]=getf(i-1);
    }
    for(int i=2;i<top-1;++i)if(getf(i)!=getf(i+1))e[++cnt]=(Line){i,i+1,S[i+1]-S[i]};
    sort(&e[1],&e[cnt+1]);
    for(int i=1;i<=cnt;++i)
        if(getf(e[i].u)!=getf(e[i].v))
            f[getf(e[i].u)]=getf(e[i].v),ans+=e[i].w;
    return ans;
}

猜你喜欢

转载自www.cnblogs.com/cjyyb/p/10462969.html