JZOJ3329. 【NOI2013模拟】树上的路径

题意

给定一棵N个结点的树，结点用正整数1..N编号，每条边有一个正整数权值。用d(a,b)表示从结点a到结点b路径上经过边的权值和，其中要求a

数据范围

对于20%的数据满足N<=1,000；
另有20%的数据满足M<=10；
对于100%的数据满足N<=50,000，M<=min{N*(N-1)/2，300,000}。

Analysis

有关树上点对的距离问题，很套路的想到点分治解决。但是题目并不是让我们求可更改的答案，那么我们在统计答案时，就不能先统计全部，再删去不合法。考虑对于每一个重心，怎么找一个合法的点对，该点对必然不在同一子树，若确定点对中的一个点，另一个点则是若干子树中的点的其中之一。然后我们考虑把每一次重心往下求$dis$的序列拉出来($dfs$序)，那么确定点对右边的点，左边的点可选择的位置是一段区间。然后考虑一个四元组$(l,r,x,pos)$，$x$可选择的点的区间左右端点。区间$dis$最大的位置，那么该四元组的权值为$dis_x+dis_{pos}$，是一个合法答案。丢入一个大根堆，每次取出一个四元组后，根据最大值位置，再分裂区间，塞入新的两个四元组。

Code

# include<cstdio>
# include<cstring>
# include<algorithm>
# include<queue>
using namespace std;
const int N = 5e4 + 5;
const int M = 3e5 + 5;
int to[N << 1],st[N],nx[N << 1],len[N << 1];
int dis[N << 4],l[N << 4],r[N << 4],log_[N << 4],mx[N << 4][25];
int siz[N],vis[N],f[N];
int n,m,tot,cnt,rt;
struct node
{
    int l,r,p,x;
    bool operator < (node r) const
    { return dis[p] + dis[x] < dis[r.p] + dis[r.x]; }
};
priority_queue <node> q;
void add(int u,int v,int w)
{
    to[++tot] = v,nx[tot] = st[u],len[tot] = w,st[u] = tot;
    to[++tot] = u,nx[tot] = st[v],len[tot] = w,st[v] = tot;
}
int Max(int a,int b) { return dis[a] > dis[b] ? a : b; }
inline void findrt(int x,int fr)
{
    siz[x] = 1,f[x] = 0;
    for (int i = st[x] ; i ; i = nx[i]) 
    {
        if (to[i] == fr || vis[to[i]]) continue;
        findrt(to[i],x); siz[x] += siz[to[i]];
        f[x] = max(f[x],siz[to[i]]);
    }
    f[x] = max(f[x],tot - siz[x]);
    if (f[rt] > f[x]) rt = x;
}
inline void getdis(int x,int sum,int fr,int li,int ri)
{
    dis[++cnt] = sum,l[cnt] = li,r[cnt] = ri;
    for (int i = st[x] ; i ; i = nx[i])
    if (to[i] != fr && !vis[to[i]]) getdis(to[i],sum + len[i],x,li,ri);
}
inline void getans(int x,int lim)
{
    vis[x] = 1; int li = ++cnt,ri = cnt;
    for (int i = st[x] ; i ; i = nx[i])
    {
        if (vis[to[i]]) continue;
        getdis(to[i],len[i],x,li,ri);
        ri = cnt;
    }
    for (int i = st[x] ; i ; i = nx[i])
    {
        if (vis[to[i]]) continue;
        tot = siz[to[i]],rt = 0;
        findrt(to[i],0); getans(rt,lim);
    }
}
int qry(int l,int r)
{
    if (l > r) return -1;
    if (!r) return 0;
    int k = log_[r - l + 1];
    return Max(mx[l][k],mx[r - (1 << k) + 1][k]);
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i = 1 ; i < n ; ++i)
    {
        int u,v,w; scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    } f[0] = N;
    tot = n,rt = 0;
    findrt(1,0),getans(rt,0); mx[1][0] = 1;
    for (int i = 2 ; i <= cnt ; ++i) mx[i][0] = i,log_[i] = log_[i >> 1] + 1;
    for (int j = 1 ; j <= log_[cnt] ; ++j)
        for (int i = 1 ; i + (1 << j) - 1 <= cnt ; ++i) mx[i][j] = Max(mx[i][j - 1],mx[i + (1 << (j - 1))][j - 1]);
    for (int i = 1 ; i <= cnt ; ++i)
    {
        int x = qry(l[i],r[i]);
        if (x) q.push((node){l[i],r[i],x,i});
    }
    for (int i = 1 ; i <= m ; ++i)
    {
        node now = q.top(); q.pop();
        printf("%d\n",dis[now.p] + dis[now.x]);
        int xl = qry(now.l,now.p - 1),xr = qry(now.p + 1,now.r);
        if (~xl) q.push((node){now.l,now.p - 1,xl,now.x});
        if (~xr) q.push((node){now.p + 1,now.r,xr,now.x});
    }
    return 0;
}