【比赛】NOIP2018 赛道修建

最小值最大,二分长度

然后判断赛道大于等于这个长度最多可以有多少条

可以贪心,对于一个点和它的一些儿子,儿子与儿子之间尽量多配(排序后一大一小),剩下的选个最长的留给自己的父亲就好了

具体实现可以用一个set(自测会被卡常,但是少爷机似乎很快)

#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
#define ft first
#define sd second
#define pb(a) push_back(a)
#define mp(a,b) std::make_pair(a,b)
#define REP(a,b,c) for(register int a=(b),a##end=(c);a<=a##end;++a)
#define DEP(a,b,c) for(register int a=(b),a##end=(c);a>=a##end;--a)
const int MAXN=50000+10;
int n,m,e,beg[MAXN],nex[MAXN<<1],to[MAXN<<1],was[MAXN<<1],cnt,now[MAXN],lcnt,val[MAXN];
std::multiset<int> S;
std::multiset<int>::iterator it,nit;
template<typename T> inline void read(T &x)
{
    T data=0,w=1;
    char ch=0;
    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
    if(ch=='-')w=-1,ch=getchar();
    while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
    x=data*w;
}
template<typename T> inline void write(T x,char ch='\0')
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
    if(ch!='\0')putchar(ch);
}
template<typename T> inline bool chkmin(T &x,T y){return y<x?(x=y,true):false;}
template<typename T> inline bool chkmax(T &x,T y){return y>x?(x=y,true):false;}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void insert(int x,int y,int z)
{
    to[++e]=y;
    nex[e]=beg[x];
    beg[x]=e;
    was[e]=z;
}
inline void dfs(int x,int f,int len)
{
    for(register int i=beg[x];i;i=nex[i])
        if(to[i]==f)continue;
        else dfs(to[i],x,len);
    for(register int i=beg[x];i;i=nex[i])
        if(to[i]==f)continue;
        else S.insert(val[to[i]]+was[i]);
    if(S.empty())return ;
    it=S.end();it--;
    while((*it)>=len)
    {
        S.erase(it);cnt++;
        if(S.empty())break;
        it=S.end(),it--;
    }
    int las=0;
    if(S.empty())return ;
    it=S.begin();
    while(it!=S.end())
    {
        nit=S.lower_bound(len-(*it));
        if(nit==S.end())chkmax(las,*it),S.erase(it);
        else if(it==nit)
        {
            nit++;
            if(nit==S.end())chkmax(las,*it),S.erase(it);
            else S.erase(it),S.erase(nit),cnt++;
        }
        else S.erase(it),S.erase(nit),cnt++;
        it=S.begin();
    }
    val[x]=las;
}
inline bool check(int len)
{
    REP(i,1,n)val[i]=0;
    cnt=0,dfs(1,0,len);
    return cnt>=m;
}
int main()
{
    freopen("track.in","r",stdin);
    freopen("track.out","w",stdout);
    read(n);read(m);
    REP(i,1,n-1)
    {
        int u,v,w;read(u);read(v);read(w);
        insert(u,v,w);insert(v,u,w);
    }
    int l=1,r=5e8,ans=1;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(check(mid))ans=mid,l=mid+1;
        else r=mid-1;
    }
    write(ans,'\n');
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/hongyj/p/10206177.html
今日推荐