Wannafly挑战赛4-dfs序&线段树|搜索&思维|BIT-树的距离

https://www.nowcoder.com/acm/contest/35/D
dfs序那个方法是标准做法。但是不知道怎么去用线段树搞。
给定一个树（根为1），然后给你 x和k，要求你求x为根的子树中 距离x大于等于k的点的 和。
一看子树，就知道是dfs序。关键是如何搞线段树orBIT ，把距离根节点的值 映射到 线段树上后（线段树的下标意义代表时间。）
维护最大值和最小值和 和（即线段树子树内的和qwq）。 还是没有特别专注这种东西。不错的题。
② 第二种方法，是dfs维护 距离根长度，x子树内的节点和x的距离和。
另外还要维护 fa（每个节点的父亲节点，因为后来用bfs计算的时候可以应用，不然就要麻烦一点）
用bfs计算节点的时候，因为递归性。
（如图，如果一个点距离x点已经大于等于k，那么他的子节点必然，所以我们如果计算好后，直接加就行），否则的话，就要累加距离，然后再进一步的递归计算。
这其实比单纯的bfs要快好多了，因为开始维护的sums数组。

ps：写的其实比较蠢笨。因为u22 这个，直接用x代替就好，累加的永远是和 x的距离qwq。u22永远只能是x。。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
/* 可以通过一个 dfs维护两个变量。
一个  节点距离根节点的 距离len[u].
另一个是  节点内的点距离 节点的距离和 sum[u].
因为我们发现了一个递归的性质，
如果一个点距离 x节点的距离大于等于k。
那么他的子节点也是。
我们可以统计一个长度再bfs里。
*/
typedef long long ll;
const int maxn=2e5*2+200;
int m;
int len;
struct Node{
      int to,next,cos;
}node[maxn];
int head[maxn];
void add(int a,int b,int c){
    node[len].to=b;
    node[len].next=head[a];
    node[len].cos=c;
    head[a]=len++;
}
ll lens[maxn];
ll sums[maxn];
int siz[maxn];
int fa[maxn];
void init(){
    memset(head,-1,sizeof(head));
    len=0;
}
void dfs(int u,int fas){
     sums[u]=0;
     siz[u]=1;//ben jiedian.
     for(int i=head[u];i!=-1;i=node[i].next){
        int to=node[i].to;
        int cost=node[i].cos;
        if(to==fas) continue;
        fa[to]=u;//这个再bfs的时候有一点用处。
        lens[to]=lens[u]+1ll*cost;
         dfs(to,u);
         siz[u]+=siz[to];
         sums[u]+=sums[to]+(1ll*siz[to]*cost);
     }
}
void bfs(int x,int k){
    ll ans=0;
    queue<pair<int,int> >q;
    q.push(make_pair(x,x));
    while(!q.empty()){
         int u=q.front().first;
         int u22=q.front().second;
         q.pop();
         for(int i=head[u];i!=-1;i=node[i].next){
             int to=node[i].to;
             int cost=node[i].cos;
             if(to==fa[u]) continue;
             if(lens[to]-lens[u22]>=1ll*k){
                ans=ans+(sums[to]+1ll*siz[to]*(lens[to]-lens[u22]));
             }
             else{
                q.push(make_pair(to,u22));
             }
         }
    }
    printf("%lld\n",ans);
}
int main()
{   int a,b,x,k,n;
    scanf("%d",&m);
     init();
     for(int i=1;i<m;i++){
         scanf("%d%d",&a,&b);
         add(a,i+1,b);
         add(i+1,a,b);
     }
     dfs(1,-1);
     /*for(int i=1;i<=m;i++){
         cout<<s7ums[i]<<" "<<siz[i]<<"??len"<<lens[i]<<endl;
     }*/
     scanf("%d",&n);
     while(n--){
          scanf("%d%d",&x,&k);
         bfs(x,k);
     }
    return 0;
}

线段树代码是别人的。还有一个bit的。 那个类写的特别骚气qwq

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 200010;
struct edge{
    int to,next;
    LL cost;
}E[maxn*2];
int head[maxn], edgecnt;
void init(){
    memset(head, -1, sizeof(head));
    edgecnt = 0;
}
void add(int u, int v, LL w){
    E[edgecnt].to = v, E[edgecnt].next = head[u], E[edgecnt].cost = w, head[u] = edgecnt++;
}
int L[maxn], R[maxn], dfs_clk;
LL a[maxn], d[maxn];
struct node{
    int l, r;
    LL minn, maxx, sum;
}Tree[maxn<<2];
void dfs(int u, int pre){
    L[u] = ++dfs_clk;
    for(int i=head[u]; ~i; i=E[i].next){
        int v = E[i].to;
        if(v == pre) continue;
        d[v] = d[u] + E[i].cost;
        dfs(v, u);
    }
    R[u] = dfs_clk;
}
void push_up(int rt){
    Tree[rt].maxx = max(Tree[rt*2].maxx, Tree[rt*2+1].maxx);
    Tree[rt].minn = min(Tree[rt*2].minn, Tree[rt*2+1].minn);
    Tree[rt].sum = Tree[rt*2].sum + Tree[rt*2+1].sum;
}
void build(int l, int r, int rt){
    Tree[rt].l = l, Tree[rt].r = r;
    if(l == r){
        Tree[rt].maxx = Tree[rt].minn = Tree[rt].sum = a[l];
        return;
    }
    int mid = l+r>>1;
    build(l, mid, rt*2);
    build(mid+1, r, rt*2+1);
    push_up(rt);
}
LL query(int L, int R, LL s, LL v, int rt){
    if(L<=Tree[rt].l && Tree[rt].r<=R){
        if(Tree[rt].minn>=v) return Tree[rt].sum - 1LL*s*(Tree[rt].r-Tree[rt].l+1);
        if(Tree[rt].maxx<v) return 0;
    }
    int mid = (Tree[rt].l+Tree[rt].r)/2;
    if(R<=mid) return query(L, R, s, v, rt*2);
    else if(L>mid) return query(L, R, s, v, rt*2+1);
    else return query(L, mid, s, v, rt*2) + query(mid+1, R, s, v, rt*2+1);
}

int main(){
    init();
    int n;
    scanf("%d", &n);
    for(int i=2; i<=n; i++){
        int p;
        LL x;
        scanf("%d %lld", &p, &x);
        add(p, i, x);
        //add(i, p, x);
    }
    dfs(1, -1);
    for(int i=1; i<=n; i++) a[L[i]] = d[i];
    build(1, n, 1);
    int q;
    scanf("%d", &q);
    while(q--){
        int x;
        LL k;
        scanf("%d %lld", &x, &k);
        LL ans = query(L[x], R[x], d[x], d[x]+k, 1);
        printf("%lld\n", ans);
    }
    return 0;
}

③别人家的BIT

#include <bits/stdc++.h>
using namespace std;

#define endl "\n"
const int N = 1e6 + 8;
int lt[N], sum, tot, n, q;
long long ans[N];
struct BIT
{
    long long a[N];
    void insert(int pos, long long val)
    {
        pos++;
        for (int i = pos; i <= n + 1; i += i & (-i)) a[i] += val;
    }
    long long sigma(int pos)
    {
        long long res = 0;
        for (int i = pos; i > 0; i -= i & (-i)) res += a[i];
        return res;
    }
    long long query(int l, int r)
    {
        l++; r++;
        if (l == 0) return sigma(r);
        return sigma(r) - sigma(l - 1);
    }
}T1, T2;
struct edge
{
    int v, w, nt;
}eg[N << 1];
struct query
{
    long long x;
    int id, w, size;
    long long dis;
    bool operator <(const query &b) const
    {
        return x < b.x;
    }
    void print()
    {
        cout << x << " " << id << " " << w << " " << size << " " << dis << endl;
    }
}Q[N];
struct node
{
    long long dis;
    int w, size;
    bool operator < (const node &b) const
    {
        return dis < b.dis;
    }
    void print()
    {
        cout << dis << " " << w << " " << size << endl;
    }
}a[N];
void add(int u, int v, int w)
{
    eg[++sum] = (edge){v, w, lt[u]}; lt[u] = sum;
}
void dfs(int u)
{
    a[u].w = ++tot; a[u].size = 1;
    for (int i = lt[u]; i; i = eg[i].nt)
    {
        a[eg[i].v].dis = a[u].dis + eg[i].w;
        dfs(eg[i].v);
        a[u].size += a[eg[i].v].size;
    }
}
void init()
{
    cin >> n;
    for (int i = 1; i <= n; ++i) lt[i] = 0;
    sum = 1; tot = 0;
    for (int i = 1; i < n; ++i)
    {
        int v, w;
        cin >> v >> w;
        add(v, i + 1, w);
    }
    dfs(1);
    cin >> q;
    for (int i = 1; i <= q; ++i)
    {
        int x, k;
        cin >> x >> k;
        Q[i].id = i;
        Q[i].x = a[x].dis + k;
        Q[i].w = a[x].w;
        Q[i].size = a[x].size;
        Q[i].dis = a[x].dis;
    }
    sort(Q + 1, Q + q + 1);
    sort(a + 1, a + n + 1);
}
void solve()
{
    for (int i = q, j = n; i >= 1; --i)
    {
        while (j > 0 && a[j].dis >= Q[i].x)
        {
            T1.insert(a[j].w, 1);
            T2.insert(a[j].w, a[j].dis);
            j--;
        }
        int num = T1.query(Q[i].w, Q[i].w + Q[i].size - 1);
        long long sum = T2.query(Q[i].w, Q[i].w + Q[i].size - 1);
        ans[Q[i].id] = sum - num * Q[i].dis;
    }
    for (int i = 1; i <= q; ++i) cout << ans[i] << endl;
}
int main()
{
    cin.sync_with_stdio(0);
    init();
    solve();
}