【Hihocoder - 1723】子树统计(线性基合并)
生活随笔
收集整理的這篇文章主要介紹了
【Hihocoder - 1723】子树统计(线性基合并)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題干:
時間限制:10000ms
單點時限:1000ms
內存限制:256MB
描述
給定一棵N個節點的有根樹,樹上每個節點有一個非負整數權重Wi。定義節點集合S={i1, i2, ……, ir}的總權重為:(⊕是異或運算)
每次詢問一棵子樹內所有可能出現的總權重數量,即令E為所詢問的子樹內節點的集合,
|T|即為可能出現的總權重數量。
輸入
第一行包含兩個整數N,Q,表示樹的節點數目和詢問數目,節點1總是這棵樹的根部。
第二行包含N-1個整數,第i個整數P_i表示i+1號節點的父親節點。數據保證Pi≤i。
第三行包含N個整數,表示每個節點的權重Wi。
第四行包含Q個整數,每個整數Qi表示詢問子樹Qi內的可能出現的總權重數量
N≤100000,Q≤100000,Wi≤260
輸出
輸出共Q行,每行包含一個整數T表示子樹Qi內可能出現的總權重數量
樣例輸入
7 3 1 2 2 1 5 5 8 4 3 1 2 4 6 2 5 1樣例輸出
8 4 16?
解題報告:
? ? dfs一遍求出以每個節點為根節點的線性基大小,設為sz,那么2^sz就是當前查詢的答案。因為查詢次數不多,里面可以帶個log,所以不需要提前預處理出來。每次直接計算sz的大小就行了。往前回溯的時候順便給合并這兩個線性基。
AC代碼:(451ms)
#include<cstdio> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<vector> #include<set> #include<string> #include<cmath> #include<cstring> #define ll long long #define pb push_back #define pm make_pair using namespace std; const int MAX = 2e5 + 5; int n,q,fa[MAX]; vector<int> vv[MAX]; const int Bit = 64; ll Base[MAX][66]; ll w[MAX],db[123]; void insert(int cur,ll x) {for(int j = Bit-1; j>=0; j--) {if(x>>j & 1) {if(Base[cur][j] == 0) {Base[cur][j] = x;break;}else x ^= Base[cur][j];}} } void merge(int cur,int rt) {for(int j = Bit-1; j>=0; j--) {if(Base[cur][j]) {insert(rt,Base[cur][j]);}} } void dfs(int cur,int rt) {int up = vv[cur].size();for(int i = 0; i<up; i++) {int v = vv[cur][i];if(v == rt) continue;dfs(v,cur);merge(v,cur);}insert(cur,w[cur]); } int main() {db[0] = 1;for(int i = 1; i<=66; i++) db[i] = db[i-1]*2;cin>>n>>q;for(int i = 2; i<=n; i++) {scanf("%d",&fa[i]);vv[fa[i]].pb(i);}for(int i = 1; i<=n; i++) scanf("%lld",w+i);dfs(1,0);while(q--) {int x;scanf("%d",&x);int sz = 0;for(int j = 0; j<=Bit-1; j++) {if(Base[x][j]) sz++;}printf("%lld\n",db[sz]);}return 0 ;}?
總結
以上是生活随笔為你收集整理的【Hihocoder - 1723】子树统计(线性基合并)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 申请信用卡邮箱怎么填 邮箱填写这两点作用
- 下一篇: 申请信用卡拼音格式 字母要大写