动态连续和查询问题,可用树状数组(二叉索引树)快速完成。
每个节点只影响它的父节点。x的父节点为Lowbit(x)。
将这个树状数组存在数组c中。原数组a对后续Add(),Sum()没什么作用。
int Sum(int x)
{
int ret = 0;
while(x > 0) {
ret += c[x];
x -= lowbit(x);
}
return ret;
}
void Add(int x, int d)
{
while(x <= n) {
c[x] += d;
x += lowbit(x);
}
}
总时间复杂度为O(nlogn)
题意:有一颗苹果树,每个节点上有一个苹果,有两种操作:
C X,修改x点处的状态(从无到有,或从有到无);
Q X,查询x点及其子树分支共有多少苹果。
思路:每个节点都可以有多个子树,我们通过dfs来给每个节点编号。每个节点保存两个值,表示它的子树范围。然后把新的节点放进树状数组中更新查询。
#include<cstdio>
#include<cstring>
#include<vector>
#define lowbit(x) (x&-x)
using namespace std;
const int maxn = 100000 + 5;
vector<vector<int> > a(maxn);
int c[maxn],l[maxn],r[maxn],tot,s[maxn];
int n;
void add(int x, int k)
{
for(int i = x; i <= n; i+=lowbit(i))
c[i] += k;
}
int sum(int x)
{
int ret = 0;
for(int i = x; i > 0; i-=lowbit(i))
ret += c[i];
return ret;
}
void dfs(int x)
{
l[x] = tot;
for(int i = 0; i < a[x].size();i++) {
tot++;
dfs(a[x][i]);
}
r[x] = tot;
}
int main()
{
int x, y, k, u, v, m;
char op[3];
while(~scanf("%d",&n)) {
memset(c,0,sizeof(c));
memset(s,0,sizeof(s));
memset(l,0,sizeof(l));
memset(r,0,sizeof(r));
for(int i = 0; i < n; i++) a[i].clear();
for(int i = 1; i < n; i++) {
scanf("%d%d",&u,&v);
a[u].push_back(v);
}
tot = 1;
dfs(1);//给每个节点编号,赋左右值
for(int i = 1; i <= n; i++) {//把每个点加到树状数组中
s[i] = 1;
add(i,1);
}
scanf("%d",&m);
while(m--) {
scanf("%s%d",op,&x);
if(op[0] == 'Q') {//查询
printf("%d\n",sum(r[x])-sum(l[x]-1));
}
else {
if(s[x]) add(l[x],-1);
else add(l[x],1);
s[x] = !s[x];
}
}
}
return 0;
}
【还有一点,vector< vector < int > > a(maxn);第一次知道后面用小括号。】