[bzoj4999][樹鏈剖分][動態開點線段樹]This Problem Is Too Simple!
阿新 • • 發佈:2018-12-12
Description
給您一顆樹,每個節點有個初始值。 現在支援以下兩種操作:
- C i x(0<=x<2^31) 表示將i節點的值改為x。
- Q i j x(0<=x<2^31) 表示詢問i節點到j節點的路徑上有多少個值為x的節點。
Input
第一行有兩個整數N,Q(1 ≤N≤ 100,000;1 ≤Q≤ 200,000),分別表示節點個數和操作個數。 下面一行N個整數,表示初始時每個節點的初始值。 接下來N-1行,每行兩個整數x,y,表示x節點與y節點之間有邊直接相連(描述一顆樹)。 接下來Q行,每行表示一個操作,操作的描述已經在題目描述中給出。
Output
對於每個Q輸出單獨一行表示所求的答案。
Sample Input
5 6
10 20 30 40 50
1 2
1 3
3 4
3 5
Q 2 3 40
C 1 40
Q 2 3 40
Q 4 5 30
C 3 10
Q 4 5 30
Sample Output
0
1
1
0
題解
離散化 對每個顏色開一棵線段樹 沒了…
#include<cstdio> #include<cstring> #include<cstdlib> #include<algorithm> #include<cmath> #include<queue> #include<vector> #include<ctime> #define LL long long #define mp(x,y) make_pair(x,y) using namespace std; inline int read() { int f=1,x=0;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } inline void write(int x) { if(x<0)putchar('-'),x=-x; if(x>9)write(x/10); putchar(x%10+'0'); } inline void print(int x){write(x);printf(" ");} struct node{int x,y,next;}a[210000];int len,last[110000]; void ins(int x,int y){len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;} int fa[110000],dep[110000],tot[110000],son[110000],n,m; void pre_tree_node(int x) { tot[x]=1; for(int k=last[x];k;k=a[k].next) { int y=a[k].y; if(y!=fa[x]) { fa[y]=x;dep[y]=dep[x]+1; pre_tree_node(y); if(tot[y]>tot[son[x]])son[x]=y; tot[x]+=tot[y]; } } } int ys[110000],top[110000],z; void pre_tree_edge(int x,int tp) { ys[x]=++z;top[x]=tp; if(son[x])pre_tree_edge(son[x],tp); for(int k=last[x];k;k=a[k].next) if(a[k].y!=fa[x]&&a[k].y!=son[x])pre_tree_edge(a[k].y,a[k].y); } struct trnode{int lc,rc,c;}tr[310000*25];int trlen,rt[310000]; void modify(int &now,int l,int r,int p,int c) { if(!now)now=++trlen; if(l==r){tr[now].c+=c;return ;} int mid=(l+r)/2; if(p<=mid)modify(tr[now].lc,l,mid,p,c); else modify(tr[now].rc,mid+1,r,p,c); tr[now].c=tr[tr[now].lc].c+tr[tr[now].rc].c; } int findsum(int now,int l,int r,int ql,int qr) { if(!now)return 0; if(l==ql&&r==qr)return tr[now].c; int mid=(l+r)/2; if(qr<=mid)return findsum(tr[now].lc,l,mid,ql,qr); else if(mid+1<=ql)return findsum(tr[now].rc,mid+1,r,ql,qr); else return findsum(tr[now].lc,l,mid,ql,mid)+findsum(tr[now].rc,mid+1,r,mid+1,qr); } struct LSnode{int y,p;}w[410000];int wlen,tt; bool cmp(LSnode n1,LSnode n2){return n1.y<n2.y;} struct ask{int l,r,c,op;}A[410000]; char ch[10]; int cal[410000]; int sol(int x,int y,int ca) { int ans=0; int tx=top[x],ty=top[y]; while(tx!=ty) { if(dep[tx]>dep[ty])swap(x,y),swap(tx,ty); ans+=findsum(rt[ca],1,n,ys[ty],ys[y]); y=fa[ty];ty=top[y]; } if(x==y)return ans+findsum(rt[ca],1,n,ys[x],ys[x]); else { if(dep[x]>dep[y])swap(x,y); return ans+findsum(rt[ca],1,n,ys[x],ys[y]); } } int main() { n=read();m=read(); for(int i=1;i<=n;i++)w[i].y=read(),w[i].p=i+m; for(int i=1;i<n;i++) { int x=read(),y=read(); ins(x,y);ins(y,x); } pre_tree_node(1); pre_tree_edge(1,1);wlen=n; for(int i=1;i<=m;i++) { scanf("%s",ch+1); A[i].l=read(); if(ch[1]=='Q')A[i].r=read(),A[i].c=read(),A[i].op=2; else A[i].c=read(),A[i].op=1; w[++wlen].y=A[i].c;w[wlen].p=i; } sort(w+1,w+1+wlen,cmp); tt=1; if(w[1].p<=m)A[w[1].p].c=1; else cal[w[1].p-m]=1; for(int i=2;i<=wlen;i++) { if(w[i].y!=w[i-1].y)tt++; if(w[i].p<=m)A[w[i].p].c=tt; else cal[w[i].p-m]=tt; } for(int i=1;i<=n;i++)modify(rt[cal[i]],1,n,ys[i],1); for(int i=1;i<=m;i++) { if(A[i].op==1) { modify(rt[cal[A[i].l]],1,n,ys[A[i].l],-1); cal[A[i].l]=A[i].c; modify(rt[cal[A[i].l]],1,n,ys[A[i].l],1); } else printf("%d\n",sol(A[i].l,A[i].r,A[i].c)); } return 0; }