标签：fir struct 题目 += bit inf 出栈 space scanf
题目大意：有一棵树根为1，刚开始每条边的权值为1, 现在有m + n - 1 个操作， A ：x y ， 将x和y相连的边权值变为1, W：x， 询问x到1路径上的权值和。
思路 ： 方法一： 用dfs序建立树状数组， 每个点入栈位置的值为1， 出栈为-1, 询问的值就是sum( l [ x ] )， 修改就将出栈，入栈的点全部变成1, 巧妙的地方在于
利用dfs序求前缀和会把不是在这条链上的边全部抵消掉。
方法二：用dfs序建立线段数， 线段数里每个点表示该点到1的权值和， 修改一条边相当于把该边下边的子树的值全部减1, 相当于区间修改。
1 #include<bits/stdc++.h> 2 #define LL long long 3 #define fi first 4 #define se second 5 #define mk make_pair 6 #define pii pair<int,int> 7 8 using namespace std; 9 10 const int N=5e5+7; 11 const int M=1e4+7; 12 const int inf=0x3f3f3f3f; 13 const LL INF=0x3f3f3f3f3f3f3f3f; 14 const int mod=1e9 + 7; 15 16 int n, m, tot, st[N], l[N], r[N]; 17 18 struct BIT { 19 int a[N]; 20 void modify(int pos, int v) { 21 for(int i = pos; i <= tot; i += i & -i) 22 a[i] += v; 23 } 24 25 int sum(int pos) { 26 int ans = 0; 27 for(int i = pos; i; i -= i & -i) 28 ans += a[i]; 29 return ans; 30 } 31 }bit; 32 33 vector<int> edge[N]; 34 void dfs(int u) { 35 l[u] = ++tot; 36 for(int i = 0; i < edge[u].size(); i++) { 37 int v = edge[u][i]; 38 dfs(v); 39 } 40 r[u] = ++tot; 41 } 42 int main() { 43 scanf("%d", &n); 44 for(int i = 1; i < n; i++) { 45 int u, v; scanf("%d%d", &u, &v); 46 if(u > v) swap(u, v); 47 edge[u].push_back(v); 48 } 49 dfs(1); 50 for(int i = 1; i <= n; i++) { 51 bit.modify(l[i], 1); 52 bit.modify(r[i], -1); 53 } 54 scanf("%d", &m); 55 m += n - 1; 56 while(m--) { 57 char s[5]; scanf("%s", s); 58 if(s[0] == ‘A‘) { 59 int u, v; scanf("%d%d", &u, &v); 60 if(u > v) swap(u, v); 61 bit.modify(l[v], -1); 62 bit.modify(r[v], 1); 63 } else { 64 int x; scanf("%d", &x); 65 printf("%d\n", bit.sum(l[x]) - 1); 66 } 67 } 68 return 0; 69 } 70 /* 71 */
标签：fir struct 题目 += bit inf 出栈 space scanf
原文地址：https://www.cnblogs.com/CJLHY/p/9005054.html
