【Hihocoder - 1723】子树统计(线性基合并)
題干:
時間限制:10000ms
單點時限:1000ms
內(nèi)存限制:256MB
描述
給定一棵N個節(jié)點的有根樹,樹上每個節(jié)點有一個非負(fù)整數(shù)權(quán)重Wi。定義節(jié)點集合S={i1, i2, ……, ir}的總權(quán)重為:(⊕是異或運算)
每次詢問一棵子樹內(nèi)所有可能出現(xiàn)的總權(quán)重數(shù)量,即令E為所詢問的子樹內(nèi)節(jié)點的集合,
|T|即為可能出現(xiàn)的總權(quán)重數(shù)量。
輸入
第一行包含兩個整數(shù)N,Q,表示樹的節(jié)點數(shù)目和詢問數(shù)目,節(jié)點1總是這棵樹的根部。
第二行包含N-1個整數(shù),第i個整數(shù)P_i表示i+1號節(jié)點的父親節(jié)點。數(shù)據(jù)保證Pi≤i。
第三行包含N個整數(shù),表示每個節(jié)點的權(quán)重Wi。
第四行包含Q個整數(shù),每個整數(shù)Qi表示詢問子樹Qi內(nèi)的可能出現(xiàn)的總權(quán)重數(shù)量
N≤100000,Q≤100000,Wi≤260
輸出
輸出共Q行,每行包含一個整數(shù)T表示子樹Qi內(nèi)可能出現(xiàn)的總權(quán)重數(shù)量
樣例輸入
7 3 1 2 2 1 5 5 8 4 3 1 2 4 6 2 5 1樣例輸出
8 4 16?
解題報告:
? ? dfs一遍求出以每個節(jié)點為根節(jié)點的線性基大小,設(shè)為sz,那么2^sz就是當(dāng)前查詢的答案。因為查詢次數(shù)不多,里面可以帶個log,所以不需要提前預(yù)處理出來。每次直接計算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 ;}?
總結(jié)
以上是生活随笔為你收集整理的【Hihocoder - 1723】子树统计(线性基合并)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 申请信用卡邮箱怎么填 邮箱填写这两点作用
- 下一篇: 【HDU - 5883】The Best