首页 > 代码库 > BZOJ 3784: 树上的路径

BZOJ 3784: 树上的路径

Description

问一棵树上前 \(k\) 大路径的边权.

Sol

边分治.

非常感谢数据没有菊花图.

为了写写边分治试试然后就开了这道题.

边分治非常好想,选一条重边,分成两部分,然后分别求最大值,对每个重边建一个堆维护一下,全局堆里存答案.

rebuild好像写的有问题啊qwq...疯狂RE不止...最后不管了,直接不重建树也能A...

rebuild我就是想的将他建成二叉树的样子,每个点的度数不超过3或4差不多就可以了.

Code

/**************************************************************    Problem: 3784    User: BeiYu    Language: C++    Result: Accepted    Time:4540 ms    Memory:26660 kb****************************************************************/ #include <bits/stdc++.h>using namespace std; #define debug(a) cout<<#a<<"="<<a<<" "#define mpr make_pairtypedef pair< int,int > pr;typedef pair< int,pr > prr;const int N = 100500; int n,k,rt,rtt;int d[N],sz[N],ud[N];int vis[N<<1]; int cnte;struct Edge { int to,w; }edge[N<<2];vector< int > g[N];priority_queue< pr > qq;priority_queue< pr > q[N];vector< int > nxt[N];vector< int > tmp[N][2]; inline int in(int x=0,char ch=getchar()) { while(ch>‘9‘ || ch<‘0‘) ch=getchar();    while(ch>=‘0‘ && ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();return x; } void AddEdge(int fr,int to,int w) { g[fr].push_back(cnte);edge[cnte++]=(Edge){ to,w }; } void Print() {    cout<<"start"<<endl;    for(int i=1;i<=n;i++) {        cout<<i<<"-->";        for(int j=0;j<(int)g[i].size();j++) cout<<"id:"<<g[i][j]<<" "<<edge[g[i][j]].to<<" ";        cout<<endl;    }    cout<<"-------------------------"<<endl;} void ReBuild(int u,int fa) {    int cnt=0,id=-1,lst=0,fst=0;    if(g[u].size()>3) for(int i=g[u].size()-1,v;i>=1;i--)         if((v=edge[g[u][i]].to)!=fa) {            if(cnt&1) {                g[n].push_back(g[u][i]);            }else {                g[++n].push_back(g[u][i]);                if(lst) {                    AddEdge(lst,n,0);                    AddEdge(n,lst,0);                }else {                    fst=n;                }lst=n;            }            cnt^=1;            edge[g[u][i]^1].to=n;            g[u].pop_back();        }else id=g[u][i],g[u].pop_back();    if(id!=-1) g[u].push_back(id);    if(fst) AddEdge(u,fst,0),AddEdge(fst,u,0);    for(int i=0;i<(int)g[u].size();i++) if(edge[g[u][i]].to!=fa) ReBuild(edge[g[u][i]].to,u);     } void GetSize(int u,int fa) {    d[u]=d[fa]+1,sz[u]=1;    for(int i=g[u].size()-1,v;~i;i--) if((v=edge[g[u][i]].to)!=fa && !vis[g[u][i]]) {        GetSize(v,u),sz[u]+=sz[v];    }}void GetRoot(int u,int nn) {    rt=-1,rtt=N;GetSize(u,u);//  for(int i=1;i<=n;i++) cout<<sz[i]<<" ";cout<<endl;//  for(int i=1;i<=n;i++) cout<<d[i]<<" ";cout<<endl;    queue< pr > q;q.push(mpr(u,u));    for(pr x;!q.empty();) {        x=q.front(),q.pop();        int u=x.first,tmp;        for(int i=0,v;i<(int)g[u].size();i++) if((v=edge[g[u][i]].to)!=x.second && !vis[g[u][i]]){            if(d[u]>d[v]) tmp=max(sz[u],nn-sz[u]);            else tmp=max(sz[v],nn-sz[v]);//          debug(u),debug(v),debug(tmp),debug(g[u][i])<<endl;            if(tmp<rtt) rtt=tmp,rt=g[u][i];            q.push(mpr(v,u));        }    }//  debug(u),debug(rt)<<endl;}void GetDep(int u,int fa,int w,vector< int > &vv) {    vv.push_back(w);    for(int i=0,v;i<(int)g[u].size();i++) if((v=edge[g[u][i]].to)!=fa && !vis[g[u][i]]) {        GetDep(v,u,w+edge[g[u][i]].w,vv);    }} void GetAns(int x,int nn) {    vis[x]=vis[x^1]=1;    int u=edge[x].to,v=edge[x^1].to;    GetDep(u,u,0,tmp[x][0]),GetDep(v,v,0,tmp[x][1]);         sort(tmp[x][0].begin(),tmp[x][0].end(),greater< int >());    sort(tmp[x][1].begin(),tmp[x][1].end(),greater< int >());     //  cout<<"---------------------"<<endl;//  debug(x),debug(nn),debug(u),debug(v)<<endl;//  for(int i=0;i<(int)tmp[x][0].size();i++) cout<<tmp[x][0][i]<<" ";cout<<endl;//  for(int i=0;i<(int)tmp[x][1].size();i++) cout<<tmp[x][1][i]<<" ";cout<<endl;          nxt[x].resize(tmp[x][0].size());    for(int i=0;i<(int)tmp[x][0].size();i++) nxt[x][i]=0;         for(int i=0;i<(int)tmp[x][0].size();i++) q[x].push(mpr(tmp[x][0][i]+tmp[x][1][0]+edge[x].w,i));     //  debug(tmp[x][0][i]+tmp[x][1][0]+edge[x].w)//  debug(q[x].top().first)<<endl;    qq.push(mpr(q[x].top().first,x));         if(d[u]<d[v]) swap(u,v);    int ss=sz[u];    GetRoot(u,ss);    if(rt!=-1) GetAns(rt,ss);    GetRoot(v,nn-ss);    if(rt!=-1) GetAns(rt,nn-ss);}void init() {    n=in(),k=in();    for(int i=1,u,v,w;i<n;i++)        u=in(),v=in(),w=in(),AddEdge(u,v,w),AddEdge(v,u,w);     //  ReBuild(1,1);     //  Print();     //  cout<<"qwq"<<endl;         GetRoot(1,n);     //  debug(rt)<<endl;         GetAns(rt,n);     }void Del(int x) {    pr r=q[x].top();    q[x].pop();    int u=r.second;    if(nxt[x][u]+1<(int)tmp[x][1].size()) nxt[x][u]++,q[x].push(mpr(tmp[x][0][u]+tmp[x][1][nxt[x][u]]+edge[x].w,u));    qq.push(mpr(q[x].top().first,x));}int Query() {    pr x=qq.top();qq.pop();    int r=x.first,u=x.second;    Del(u);         return r;}int main() {//  freopen("in.in","r",stdin);//  ios::sync_with_stdio(false);    init();         for(;k--;) {        printf("%d\n",Query());    }         return 0;}

  

BZOJ 3784: 树上的路径