2019河南省ccpc大小接近的点对（主席数+离散化+dfs序

最新推荐文章于 2022-06-05 15:53:27 发布

FriendA831143

最新推荐文章于 2022-06-05 15:53:27 发布

阅读量174

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/FriendA831143/article/details/111083638

本文介绍了一种使用树状数组进行区间更新查询的算法实现，该算法能够有效地处理树形结构上的区间加权操作和查询任务。通过预处理和递归方式构建数据结构，并采用离散化等技巧优化效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<string>
#include<cstring>
#include<bitset> 
#include<vector>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<iomanip>
#include<algorithm>
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define endl "\n"
#define int long long
#define PI  acos(-1)
using namespace std;

const int INF = 0x3f3f3f3f;
const int maxn = 4e5+5;
const int mod = 1e9 +7;
int n,m,k;
int a[maxn],e[maxn],ne[maxn],h[maxn],cur,idx,edt;
int root[maxn];
int d[maxn];
vector<int> num;
struct node{
	int l,r;
	int cnt;
}tr[maxn * 20];
int cnt[maxn];
void pushup(int u){
	tr[u].cnt = tr[tr[u].l].cnt + tr[tr[u].r].cnt;
}

int get(int x){
	return lower_bound(num.begin(),num.end(),x) - num.begin();
}

void add(int a,int b){
	e[cur] = b,ne[cur] = h[a],h[a] = cur++; 
}

int build(int l,int r){
	int p = ++idx;
	if(l == r) return p;
	int mid = l + r >> 1;
	tr[p].l = build(l,mid);
	tr[p].r = build(mid + 1,r);
	return p;
}

int insert(int x,int l,int r,int p){
	int q = ++idx;
	tr[q] = tr[p];
	if(l == r){
		tr[q].cnt++;
		return q;
	}
	int mid = l + r >> 1;
	if(x <= mid) tr[q].l = insert(x,l,mid,tr[p].l);
	else tr[q].r = insert(x,mid + 1,r,tr[p].r);
	pushup(q);
	return q;
}
int query(int l,int r,int L,int R,int q,int p){
	if(L >= l && R <= r) return tr[q].cnt - tr[p].cnt;
	int mid = L + R >> 1;
	int cnt = 0;
	if(l <= mid) cnt += query(l,r,L,mid,tr[q].l,tr[p].l);
	if(r >  mid) cnt += query(l,r,mid + 1,R,tr[q].r,tr[p].r);
	return cnt;
}

void dfs(int u){
	int ds = 0;
	for(int i = h[u] ; i != -1 ; i = ne[i]){
		int j = e[i];
		dfs(j);
		ds += d[j];
		cnt[u] += cnt[j];
	}
	d[u] = ds + 1;
	root[++edt] = insert(get(a[u]),0,num.size() - 1,root[edt - 1]);
	cnt[u] += query(get(a[u] - k),get(a[u] + k),0,num.size() - 1,root[edt],root[edt - d[u]]);
	
}

void solve(){
	scanf("%lld%lld",&n,&k);
	for(int i = 1 ; i <= n ; ++i ){
		scanf("%lld",&a[i]);
		num.push_back(a[i]);
		num.push_back(a[i] - k);
		num.push_back(a[i] + k);
	}
	memset(h,-1,sizeof(h));
	int x;
	for(int i = 1 ; i < n ; ++i){
		scanf("%lld",&x);
		add(x,i + 1);
	}
	sort(num.begin(),num.end());
	num.erase(unique(num.begin(),num.end()),num.end());
	root[0] = build(0,num.size() - 1);
	dfs(1);
	for(int i = 1 ; i <= n ; ++i ) printf("%lld\n",cnt[i]);
}

signed main(){
	solve();
	return 0;
}
/*
*
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*/