程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> zoj 3686 A Simple Tree Problem

zoj 3686 A Simple Tree Problem

編輯:C++入門知識

 

題目大意:

一棵樹有n個結點(1<=N<=100000),根為1,每個結點有一個標簽,初始值為0.

現在有2種操作

o操作:(o,x)對結點x及其所有子孫結點的標簽值取反(和1異或)。

q操作:(q,x)輸出結點x及其所以子孫結點的標簽值為1的個數。

操作次數為m次(1<=M<=10000)。

 

題目思路:

從根開始中序遍歷每一個結點,給每一個結點按照遍歷順序重新編號(beg)。

這樣子可以使得重新編號的結點,根及其連接的子樹的結點是連續的,這樣我們就可以操作時區間更新和區間詢問了。

並且遍歷時記錄下某根子樹中最大的結點編號(end),這樣我們更新或詢問某結點及其子樹的范圍即

[ beg[x] , end[x] ]。

 

代碼:

[cpp] 
#include <stdlib.h>  
#include <string.h>  
#include <stdio.h>  
#include <ctype.h>  
#include <math.h>  
#include <stack>  
#include <queue>  
#include <map>  
#include <set>  
#include <vector>  
#include <string>  
#include <iostream>  
#include <algorithm>  
using namespace std; 
 
#define ll __int64  
#define ls rt<<1  
#define rs ls|1  
#define lson l,mid,ls  
#define rson mid+1,r,rs  
#define middle (l+r)>>1  
#define clr_all(x,c) memset(x,c,sizeof(x))  
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))  
#define eps (1e-8)  
#define MOD 1000000007  
#define INF 0x3f3f3f3f  
#define PI (acos(-1.0))  
#pragma comment(linker, "/STACK:102400000,102400000")  
template <class T> T _max(T x,T y){return x>y? x:y;} 
template <class T> T _min(T x,T y){return x<y? x:y;} 
template <class T> T _abs(T x){return (x < 0)? -x:x;} 
template <class T> T _mod(T x,T y){return (x > 0)? x%y:((x%y)+y)%y;} 
template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;} 
template <class T> void getmax(T& x,T y){x=(y > x)? y:x;} 
template <class T> void getmin(T& x,T y){x=(x<0 || y<x)? y:x;} 
int TS,cas=1; 
const int M=100000+5; 
int n,m; 
struct node{ 
    int beg,end; 
}mp[M]; 
int fa[M],nxt[M],vec[M],tot; 
void dfs(int rt){ 
    mp[rt].beg=++tot; 
    for(int i=fa[rt];i!=-1;i=nxt[i]) dfs(vec[i]); 
    mp[rt].end=tot; 

int cov[M<<2],sum[M<<2]; 
void build(int l,int r,int rt){ 
    cov[rt]=sum[rt]=0; 
    if(l==r) return; 
    int mid=middle; 
    build(lson),build(rson); 

void pushDown(int l,int mid,int r,int rt){ 
    if(!cov[rt]) return; 
    sum[ls]=(mid-l+1)-sum[ls],cov[ls]^=1; 
    sum[rs]=(r-mid)-sum[rs],cov[rs]^=1; 
    cov[rt]=0; 

void pushUp(int rt){ 
    sum[rt]=sum[ls]+sum[rs]; 

void update(int l,int r,int rt,int L,int R){ 
    if(L<=l && r<=R){ 
        sum[rt]=(r-l+1)-sum[rt]; 
        cov[rt]^=1; 
        return; 
    } 
    int mid=middle; 
    pushDown(l,mid,r,rt); 
    if(R<=mid) update(lson,L,R); 
    else if(mid<L) update(rson,L,R); 
    else update(lson,L,mid),update(rson,mid+1,R); 
    pushUp(rt); 

int query(int l,int r,int rt,int L,int R){ 
    if(L<=l && r<=R) return sum[rt]; 
    int mid=middle; 
    pushDown(l,mid,r,rt); 
    if(R<=mid) return query(lson,L,R); 
    else if(mid<L) return query(rson,L,R); 
    else return query(lson,L,mid)+query(rson,mid+1,R); 

 
void run(){ 
    int i,j; 
    clr(fa,-1,n+1); 
    tot=0; 
    for(i=2;i<=n;i++){ 
        scanf("%d",&j); 
        nxt[tot]=fa[j],vec[tot]=i,fa[j]=tot++; 
    } 
    tot=0; 
    dfs(1); 
    build(1,n,1); 
    while(m--){ 
        char op[3]; 
        scanf("%s%d",op,&j); 
        if(op[0] == 'o') update(1,n,1,mp[j].beg,mp[j].end); 
        else printf("%d\n",query(1,n,1,mp[j].beg,mp[j].end)); 
    } 
    puts(""); 

 
void preSof(){ 

 
int main(){ 
    //freopen("input.txt","r",stdin);  
    //freopen("output.txt","w",stdout);  
    preSof(); 
    //run();  
    while((~scanf("%d%d",&n,&m))) run(); 
     //for(scanf("%d",&TS);cas<=TS;cas++) run();  
    return 0; 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved