CF #495 Div.2 E. Sonya and Ice Cream //树的直径的中心 [模板]

最新推荐文章于 2019-10-04 22:09:05 发布

RSHS

最新推荐文章于 2019-10-04 22:09:05 发布

阅读量232

点赞数

本文链接：https://blog.csdn.net/qq_41730604/article/details/101784164

版权

题目

题意

给一颗带边权的树，你可以选择连续的 $k$ 个顶点的一条链，链上的顶点都开商店，问怎么使得每一个点到最近的商店的最大值最小。

思路

这里考虑树的直径的中心，肯定是开商店的。（我不会证明，感觉是对的。。）
然后确定了这一个顶点，那么以这个顶点拎起来成为一棵树，先选择根的儿子到叶子节点最大的两个，然后加入优先队列，然后以这两个最大儿子为起点，不停的向深度更深的儿子延申。

求树的直径的中心：

1.先求直径，两次 $b f s$ ，第一次从任意一点 $x$ $b f s$ 找到离这个点最远的点 $y$ ，再从 $y$ $b f s$ 到最远的点 $z$ ，那么 $y$ 和 $z$ 就是直径的两个端点，然后中途记录前驱距离什么什么的，就可以找到想要的中点(有时候不能找到确切的中点，那就取两个离中点最近的点)。
2.树形 $d p$ +换根，找到离叶子节点最大值最小的那个顶点，就是中心。
3.树形 $d p$ 记录到叶子节点距离的最大值和次大值，就可以枚举每一个点作为转折点的最长链，从而找到直径，记录每个节点到父亲节点的距离还有编号，就可以找到中心。（因为每个点如果链向父亲节点走的话，那可以归纳到父亲节点作为转折点的链，所以一定可以找到）。

方法1：

/*   Author : Rshs
 *   Data : 2019-09-29-15.28
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,LL>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-10;
const LL mod = 1e9+7;
const int MXN = 1e6+5;
LL fd[MXN],md[MXN],dis[MXN],pa[MXN],pd[MXN],vis[MXN];
vector<PII>g[MXN];
int RT,RTT;//两个中心
void dfs(int nov,int pr){
    for(auto i:g[nov]){
        if(i.FI==pr)continue;
        fd[i.FI]=i.SE;
        dfs(i.FI,nov);
        md[nov]=max(md[nov],md[i.FI]+i.SE);
    }
}
void bfs(int n){
    queue<int>q;
    for(int i=1;i<=n;i++) dis[i]=LLONG_MAX/10;
    dis[1]=0;q.push(1);
    while(!q.empty()){
        int now=q.front();q.pop();
        for(auto i:g[now]){
            if(dis[now]+i.SE<dis[i.FI])dis[i.FI]=dis[now]+i.SE,q.push(i.FI);
        }
    }
    int mx=0,u;
    for(int i=1;i<=n;i++) if(dis[i]>mx)mx=dis[i],u=i;
    for(int i=1;i<=n;i++) dis[i]=LLONG_MAX/10;
    dis[u]=0;q.push(u);pa[u]=-1,pd[u]=0;
    while(!q.empty()){
        int now=q.front();q.pop();
        for(auto i:g[now]){
            if(dis[now]+i.SE<dis[i.FI])dis[i.FI]=dis[now]+i.SE,q.push(i.FI),pa[i.FI]=now,pd[i.FI]=i.SE;
        }
    }
    int MX=0,U;
    for(int i=1;i<=n;i++) if(dis[i]>MX)MX=dis[i],U=i;
    int DIS=0;
    for(int i=U;i!=u;i=pa[i]){
        DIS+=pd[i];
    }
    DIS/=2;
    int ddd=0;
    for(int i=U;i!=u;i=pa[i]){
        if(ddd+pd[i]>DIS) {RT=i;RTT=pa[i];break;} //找中点
        ddd+=pd[i];
    }
    if(RTT==-1) RTT=1;
}
int FMAX(int x,int xx){
    LL mx=-1;int u=-1;
    for(auto i:g[x]){
        if(i.FI==xx)continue;
        if(vis[i.FI])continue;
        if(i.SE+md[i.FI]>mx) mx=i.SE+md[i.FI],u=i.FI;
    }
    return u;
}
int work(int k,int rt,int n){
    memset(vis,0,sizeof(vis));
    memset(md,0,sizeof(md));
    memset(fd,0,sizeof(fd));
    dfs(rt,0);
    priority_queue<pair<LL,int> >pq;
    k--;vis[rt]=1;
    int u=FMAX(RT,-1);
    if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    u=FMAX(rt,u);
    if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    while(!pq.empty()&&k){
        auto now=pq.top();pq.pop();
        vis[now.SE]=1;k--;
        int u=FMAX(now.SE,-1);
        if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    }
    LL ans=0;
    for(int i=1;i<=n;i++){
        if(vis[i])continue;
        ans=max(ans,fd[i]+md[i]);
    }
    return ans;
}
int main(){
    int n,k;cin>>n>>k;
    if(n==1)return puts("0"),0;
    for(int i=1;i<n;i++){
        int sa,sb,sc;scanf("%d %d %d",&sa,&sb,&sc);
        g[sa].push_back(MP(sb,sc));
        g[sb].push_back(MP(sa,sc));
    }
    bfs(n);
    cout<<min(work(k,RT,n),work(k,RTT,n));
    return 0;
}

方法2：

 
 /*   Author : Rshs
 *   Data : 2019-09-29-15.28
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,LL>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-10;
const LL mod = 1e9+7;
const int MXN = 1e6+5;
LL md[MXN],fd[MXN];
int vis[MXN];
vector<PII>g[MXN];
 
LL MIDIS=LLONG_MAX,RT;
void dfs(int nov,int pr){
    for(auto i:g[nov]){
        if(i.FI==pr)continue;
        fd[i.FI]=i.SE;
        dfs(i.FI,nov);
        md[nov]=max(md[nov],md[i.FI]+i.SE);
    }
}
void hg(int nov,int pr){
    if(md[nov]<MIDIS)MIDIS=md[nov],RT=nov;
    /****************************************************************************/
   vector<LL>suf,pa; pa.push_back(LLONG_MIN);suf.push_back(LLONG_MIN);
    for(auto i:g[nov]) pa.push_back(max(md[i.FI]+i.SE,pa.back()));
    reverse(g[nov].begin(),g[nov].end());
    for(auto i:g[nov]) suf.push_back(max(md[i.FI]+i.SE,suf.back()));
    reverse(g[nov].begin(),g[nov].end());
    /****************************************************************************/
    for(int i=0;i<SZ(g[nov]);i++){
        auto now=g[nov][i];
        if(now.FI==pr)continue;
        LL aa=md[nov],bb=md[now.FI];  //����
        md[nov]=0;
        if(i>0)md[nov]=max(md[nov],pa[i]);
        if(i<SZ(g[nov])-1) md[nov]=max(md[nov],suf[SZ(g[nov])-i-1]);
        md[now.FI]=max(md[now.FI],md[nov]+now.SE);
        hg(now.FI,nov);
        md[nov]=aa,md[now.FI]=bb;
    }
}
int FMAX(int x,int xx){
    LL mx=-1;int u=-1;
    for(auto i:g[x]){
        if(i.FI==xx)continue;
        if(vis[i.FI])continue;
        if(i.SE+md[i.FI]>mx) mx=i.SE+md[i.FI],u=i.FI;
    }
    return u;
}
int main(){
    int n,k;cin>>n>>k;
    for(int i=1;i<n;i++){
        int sa,sb,sc;scanf("%d %d %d",&sa,&sb,&sc);
        g[sa].push_back(MP(sb,sc));
        g[sb].push_back(MP(sa,sc));
    }
    dfs(1,0);
    hg(1,0);
    memset(md,0,sizeof(md));
    dfs(RT,0);
    priority_queue<pair<LL,int> >pq;
    k--;vis[RT]=1;
    int u=FMAX(RT,-1);
    if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    u=FMAX(RT,u);
    if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    while(!pq.empty()&&k){
        auto now=pq.top();pq.pop();
        vis[now.SE]=1;k--;
        int u=FMAX(now.SE,-1);
        if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    }
    LL ans=0;
    for(int i=1;i<=n;i++){
        if(vis[i])continue;
        ans=max(ans,fd[i]+md[i]);
    }
    cout<<ans;
    return 0;
}

方法3：

/*   Author : Rshs
 *   Data : 2019-09-29-15.28
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,LL>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-10;
const LL mod = 1e9+7;
const int MXN = 1e6+5;
LL fd[MXN],md[MXN],vis[MXN];
LL mx[MXN],mx2[MXN],pos[MXN],pos2[MXN],fa[MXN];
vector<PII>g[MXN];
int RT,RTT;
void dfs(int nov,int pr){
    for(auto i:g[nov]){
        if(i.FI==pr)continue;
        fd[i.FI]=i.SE;fa[i.FI]=nov;
        dfs(i.FI,nov);
        md[nov]=max(md[nov],md[i.FI]+i.SE);
    }
    mx[nov]=mx2[nov]=0,pos[nov]=pos2[nov]=nov;
    for(auto i:g[nov]){
        if(i.FI==pr)continue;
        if(md[i.FI]+i.SE>mx[nov]){
            mx2[nov]=mx[nov];mx[nov]=md[i.FI]+i.SE;
            pos2[nov]=pos[nov] ;pos[nov]=pos[i.FI];
        }
        else if(md[i.FI]+i.SE>mx2[nov]){
            mx2[nov]=md[i.FI]+i.SE;pos2[nov]=pos[i.FI];
        }
    }
}
void getRt(int n){
    dfs(1,-1);
    int mm=0,u;
    for(int i=1;i<=n;i++) if(mx[i]+mx2[i]>mm) mm=mx[i]+mx2[i],u=i;
    vector<int>v,vv,c,cc;
    for(int i=pos[u];i!=u;i=fa[i]){
        v.push_back(fa[i]),c.push_back(fd[i]);
    }
    if(mx2[u]){
        for(int i=pos2[u];i!=u;i=fa[i]){
            vv.push_back(i),cc.push_back(fd[i]);
        }
    }
    reverse(vv.begin(),vv.end());reverse(cc.begin(),cc.end());
    for(auto i:vv) v.push_back(i);
    for(auto i:cc) c.push_back(i);
    LL DIS=0,ddd=0;
    for(auto i:c) DIS+=i;
    DIS/=2;
    for(int i=0;i<SZ(c);i++){
        if(ddd+c[i]>DIS) {
            RT=v[i];
            if(i>0)RTT=v[i-1];
            else RTT=1;
            break;
        }
        ddd+=c[i];
    }
}
 
int FMAX(int x,int xx){
    LL mx=-1;int u=-1;
    for(auto i:g[x]){
        if(i.FI==xx)continue;
        if(vis[i.FI])continue;
        if(i.SE+md[i.FI]>mx) mx=i.SE+md[i.FI],u=i.FI;
    }
    return u;
}
int work(int k,int rt,int n){
    memset(vis,0,sizeof(vis));
    memset(md,0,sizeof(md));
    memset(fd,0,sizeof(fd));
    dfs(rt,0);
    priority_queue<pair<LL,int> >pq;
    k--;vis[rt]=1;
    int u=FMAX(RT,-1);
    if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    u=FMAX(rt,u);
    if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    while(!pq.empty()&&k){
        auto now=pq.top();pq.pop();
        vis[now.SE]=1;k--;
        int u=FMAX(now.SE,-1);
        if(u!=-1)  pq.push(MP(md[u]+fd[u],u));
    }
    LL ans=0;
    for(int i=1;i<=n;i++){
        if(vis[i])continue;
        ans=max(ans,fd[i]+md[i]);
    }
    return ans;
}
int main(){
    int n,k;cin>>n>>k;
    if(n==1)return puts("0"),0;
    for(int i=1;i<n;i++){
        int sa,sb,sc;scanf("%d %d %d",&sa,&sb,&sc);
        g[sa].push_back(MP(sb,sc));
        g[sb].push_back(MP(sa,sc));
    }
    getRt(n);
    cout<<min(work(k,RT,n),work(k,RTT,n));
    return 0;
}