POJ3321 树状数组,多叉树转线性结构,经典基础题

1 题意

对于一个多叉树的,有两种操作,在某个点上摘一个苹果或者生成一个苹果(每个点要么有一个苹果要么有0个苹果);输出某个点作为root的子树上的苹果之和(包括root)。

2 分析

将多叉树用邻接表存储下来,通过DFS将多叉树转换为线性结构并且记录子树所在区间,然后构造树状数组,进行改动和查询。

3

#include 
#include 
#include 
#include 
using namespace std;

const int maxn=100010;
//存储边:
struct Edge{
    int next;
    int from;
    int to;
}edge[maxn];
int head[maxn];
int cur,cnt;//cur-用于存储边,cnt-用于转变为线性结构
int n,m;
//记录线性结构+记录子树:
struct Apple{
    int left;
    int right;//apple[x].right实际上是结点x从树结构转变为线性结构的一个映射
}apple[maxn];

//树状数组存储结构:
int sum[maxn];//前缀和,sum[x,y]=sum[y]-sum[x-1];
int a[maxn];//结点的权值

void Ini(){
    cur=0;
    cnt=1;
    memset(head,-1,sizeof(head[0])*(n+1));
    memset(sum,0,sizeof(sum[0])*(n+1));
    memset(apple,0,sizeof(apple[0])*(n+1));
}
void Add(int from,int to){
    edge[cur].next=head[from];
    edge[cur].from=from;
    edge[cur].to=to;
    head[from]=cur++;
}
void Dfs(int u){
    apple[u].left=cnt;
    for(int i=head[u];i!=-1;i=edge[i].next){
        Dfs(edge[i].to);
    }
    apple[u].right=cnt++;
}
int lowbit(int x){
    return x&(-x);//x的二进制表示中,从最右边数第一个不为0的位数,返回2^x
}
void Change(int x,int value){
    for(int i=x;i0;i-=lowbit(i)){
        res+=sum[i];
    }
    return res;
}
void Build(int value){
    for(int i=1;i<=n;i++){
        a[i]=1;
        Change(i,1);
    }
}
int main()
{
    int num1,num2;
    char str[3];
    while(~scanf("%d",&n)){
        Ini();
        for(int i=1;i

你可能感兴趣的:(ACM,数据结构)