当前位置: 代码迷 >> 综合 >> noip 2018 洛谷 P5021 赛道修建
  详细解决方案

noip 2018 洛谷 P5021 赛道修建

热度:30   发布时间:2023-12-06 07:31:51.0

题目:赛道修建

思路:
二分答案。
judge时,令节点1为根节点,dfs求解。
安利

代码:

#include<bits/stdc++.h>
using namespace std;#define maxn 50000
#define read(x) scanf("%d",&x)int cnt[maxn+5],f[maxn+5];struct Edge{
    int u,v,w,f;Edge(){
    }Edge(int uu,int vv,int ww) {
    u=uu,v=vv,w=ww;}Edge(Edge oth,int ff) {
    u=oth.u,v=oth.v,w=oth.w,f=ff;}bool operator < (const Edge& oth) const {
    return f<oth.f;}
};int n,m;vector<Edge> g[maxn+5];vector<Edge> chd[maxn+5];void dfs(int x,int fa,int mid) {
    cnt[x]=f[x]=0;chd[x].clear();for(int i=0;i<g[x].size();i++) {
    Edge y=g[x][i];if(y.v==fa) continue;dfs(y.v,x,mid);f[y.v]+=y.w;if(f[y.v]>=mid) cnt[x]++;else chd[x].push_back(Edge(y,f[y.v]));}if(chd[x].size()==0) {
    return ;}sort(chd[x].begin(),chd[x].end());int L=0,R=chd[x].size()-1,mm=-1,dd=0;while(L<R) {
    if(chd[x][L].f+chd[x][R].f>=mid) cnt[x]++,dd++,L++,R--;else L++;		}L=0,R=chd[x].size()-1;while(L<=R) {
    int Mid=(L+R)/2;int flg=0;int l=0,r=chd[x].size()-1;if(l==Mid) l++;if(r==Mid) r--;while(l<r) {
    if(chd[x][l].f+chd[x][r].f>=mid) flg++,l++,r--;else l++;if(l==Mid) l++;if(r==Mid) r--;}if(flg==dd) L=Mid+1,mm=Mid;else R=Mid-1;}if(mm!=-1) f[x]=max(f[x],chd[x][mm].f);return ;
}bool judge(int mid) {
    int ans=0;dfs(1,0,mid);for(int i=1;i<=n;i++) ans+=cnt[i];return ans>=m;
}int main() {
    int L=0,R=0;read(n),read(m);for(int i=1;i<n;i++) {
    int x,y,z;read(x),read(y),read(z);R+=z;g[x].push_back(Edge(x,y,z)),g[y].push_back(Edge(y,x,z));}R/=m;int ans;while(L<=R) {
    int mid=(L+R)/2;if(judge(mid)) L=mid+1,ans=mid;else R=mid-1;}printf("%d",ans);return 0;
}
  相关解决方案