poj 3321/3263 树状数组(苹果树/排身高)

题意:每个分叉点及末梢可能有苹果(最多1个), 每次可以摘掉一个苹果,或有一个苹果新长出来,随时查询某个分叉点所在的子树里, 一共有多少个苹果。

思路:深度优先遍历整个苹果树,为每个节点标记一个开始时间和结束时间(所有时间都不相同),显然子树里面所有节点的开始和结束时间,都位于子树树根的开始和结束时间之间。那么苹果的有无只需在开始时间那个点(在结束时间那个点上也可以)上面进行加减1的操作即可。然后每次查询就看看从开始点到结束点之间的和一共是多少。用线段树或者树状数组皆可。树状数组容易写,遂用之。

#include 
#include 
#include 
#include 
#define N 100005
#define INF 0x3fffffff
using namespace std;
#define N 100005
int tree[N<<1];
struct edge{
    int y,next;
}e[N<<1];
int first[N],top,n,q,len;
int s[N],w[N],flag[N];
void add(int x,int y){
    e[top].y = y;
    e[top].next = first[x];
    first[x] = top++;
}
void dfs(int x){//对树进行dfs,得到起始和终止的时间戳
    int i;
    s[x] = ++len;
    for(i = first[x];i!=-1;i=e[i].next)
        if(!s[e[i].y])
            dfs(e[i].y);
    w[x] = ++len;
}
int lowbit(int x){
    return x&(-x);
}
void addtree(int i,int x){
    for(int j = i;j<=2*n;j+=lowbit(j))
        tree[j] += x;
}
int sum(int x){
    int i,res = 0;
    for(i = x;i>=1;i-=lowbit(i))
        res += tree[i];
    return res;
}
int main(){
    int i,a,b,x;
    char ch;
    scanf("%d",&n);
    memset(s,0,sizeof(s));
    memset(first,-1,sizeof(first));
    top = len = 0;
    for(i = 1;i

poj3263:计数的思路如出一辙。需要注意的是a看到b和b看到a是一个意思,不能重复计数,所以要筛一遍。另外题目给的最高高度的索引是没有用的。

#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define clc(s,t) memset(s,t,sizeof(s))
#define INF 0x3fffffff
#define N 10005
struct node{
    int a,b;
}s[N];
int tree[N];
int n,m,h,t;
int cmp(node x,node y){
    if(x.a == y.a)
        return x.b=1;j-=lowbit(j))
        res += tree[j];
    return res;
}
int main(){
    int i,j,x,y;
    scanf("%d %d %d %d",&n,&t,&h,&m);
    for(i = 0;i s[i].b)
            swap(s[i].a,s[i].b);
    }
    sort(s,s+m,cmp);
    for(i = 0;i


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