树的距离【可持久化线段树】【Wannafly挑战赛4】

题目链接


题目描述

wyf非常喜欢树。一棵有根数树上有N个节点,1号点是他的根,每条边都有一个距离,而wyf是个爱问奇怪问题的熊孩子,他想知道对于某个点x,以x为根的子树上,所有与x距离大于等于k的点与x的距离之和。

输入描述:

 

第一行一个正整数N

接下来N-1描述这棵树,每行两个数第i行两个数p和D表示树上有一条p到i+1长度为D的边。(p<=i)

下面一行一个正整数Q表示wyf的询问次数。

接下来Q行每行两个正整数x和k。 (1<=N,Q<=2x105,1<=D,K<=106)

输出描述:

对于每次询问x,k输出以x为根的子树上,所有与x距离大于等于k的点与x的距离之和。(若不存在这样的点,则输出应为0)

  对于这样的一颗树,我们查询的是以x为根的子树的所有距离"≥K"的子节点的值的权值的和。

  思路:对于这样的点,我们从根节点1开始向下跑,建立时间戳来更新所有的节点,时间戳表明的是对应查到的节点的对应的时间上的节点的可持久化线段树的状态。然后,千万不要忘记使用long long,WA到自闭…… 


#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define efs 1e-8
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 2e5 + 7;
const ll _UP = 2e11;
int N, Q, head[maxN], cnt, root[maxN], lc[50 * maxN], rc[50 * maxN], tot, id[maxN], Index, ou[maxN];
ll sum[50 * maxN], pot[50 * maxN], dis[maxN];
struct Eddge
{
    int nex, to, cost;
    Eddge(int a=-1, int b=0, int c=0):nex(a), to(b), cost(c) {}
}edge[maxN];
inline void addEddge(int u, int v, int w)
{
    edge[cnt] = Eddge(head[u], v, w);
    head[u] = cnt++;
}
inline void pushup(int rt)
{
    sum[rt] = sum[lc[rt]] + sum[rc[rt]];
    pot[rt] = pot[lc[rt]] + pot[rc[rt]];
}
inline void insert(int &rt, int old, ll l, ll r, ll qx)
{
    rt = ++tot;
    lc[rt] = lc[old]; rc[rt] = rc[old];
    if(l == r)
    {
        sum[rt] = sum[old] + qx;
        pot[rt] = pot[old] + 1;
        return;
    }
    ll mid = HalF;
    if(qx <= mid) insert(lc[rt], lc[old], l, mid, qx);
    else insert(rc[rt], rc[old], mid + 1, r, qx);
    pushup(rt);
}
inline void Query(int rt, int old, ll l, ll r, ll qx, ll &node, ll &insum)
{
    if(!rt) return;
    if(qx <= l) { node += pot[rt] - pot[old]; insum += sum[rt] - sum[old]; return; }
    ll mid = HalF;
    if(qx <= mid) { node += pot[rc[rt]] - pot[rc[old]]; insum += sum[rc[rt]] - sum[rc[old]];  Query(lc[rt], lc[old], l, mid, qx, node, insum); }
    else Query(rc[rt], rc[old], mid + 1, r, qx, node, insum);
}
inline void dfs(int u, ll dist)
{
    dis[u] = dist;
    id[u] = ++Index;
    insert(root[Index], root[Index - 1], 1, _UP, dis[u]);
    for(int i=head[u], v, c; ~i; i=edge[i].nex)
    {
        v = edge[i].to; c = edge[i].cost;
        dfs(v, dist + c);
    }
    ou[u] = Index;
}
inline void init()
{
    cnt = tot = Index = 0;
    memset(head, -1, sizeof(head));
    memset(root, 0, sizeof(root));
    sum[0] = pot[0] = 0;
}
int main()
{
    scanf("%d", &N);
    init();
    for(int i=2, u, w; i<=N; i++)
    {
        scanf("%d%d", &u, &w);
        addEddge(u, i, w);
    }
    dfs(1, 0);
    scanf("%d", &Q);
    int u;
    ll sum_point, sum_num, k;
    while(Q--)
    {
        scanf("%d%lld", &u, &k);
        sum_point = sum_num = 0;
        Query(root[ou[u]], root[id[u]], 1, _UP, dis[u] + k, sum_point, sum_num);
        printf("%lld\n", sum_num - sum_point * dis[u]);
    }
    return 0;
}
/*
6
1 2
1 3
2 1
2 2
3 1
1
6 1
ans:0
*/

/*
8
1 5
1 4
1 3
2 1
2 2
3 2
4 3
1
4 1
ans:3
*/

 

你可能感兴趣的:(线段树,数据结构)