Description

Input

Output

Sample Input

10 5 6
1 2 3 4 5 6 7 8 9 10
9 7
7 10
6 5
7 5
5 8
5 1
2 1
3 2
2 4
2 3 4
2 9 5
1 7 3
4 8 2
5 6 6
2 3
2 5
1 4
3 5
5 1
6 1

Sample Output

0
43
4
27
11
13

Data Constraint

Hint

Solution

題意：

• 給定一棵以 1 為根的樹，樹上結點上有一些數字。

• 這些數字每過單位 1 時間會走到其所在結點的父親上，結點 1 上的數字會消失。

• 有若干詢問，問某一時間一棵子樹中的數字和。

思路：

• 我們考慮一個數字會被一個詢問統計到是一種什么情況：

• 設這個數字在 txtx 時間出現在結點 xx 上，然后會被結點 yy 在 tyty 時間統計，

• 那么就會滿足不等式：

  ① dep[x]?dep[y]≥ty?txdep[x]?dep[y]≥ty?tx 且 tx≤tytx≤ty ；
  ②xx 必須在 yy 的子樹內；

• 其中 dep[i]dep[i] 表示結點 ii 的深度。

• 考慮①：我們移項，就會有 dep[x]+tx≥dep[y]+tydep[x]+tx≥dep[y]+ty，

• 顯然左邊只會和 xx 有關，右邊只會和 yy 有關；

• 考慮②：我們考慮使用 dfsdfs 序，

• 那么②等價于 dfn[x]≥dfn[y]dfn[x]≥dfn[y] 且 dfn[x]<dfn[x]+size[x]dfn[x]<dfn[x]+size[x]，

• 其中 dfn[i]dfn[i] 表示 ii 的 dfsdfs 序，

• Size[i]Size[i] 表示以 ii 為子樹的結點個數。

• 綜上，我們把題目中的第 xx 數字寫成二維平面上的一個帶權點，

• 橫坐標為 dep[x]+txdep[x]+tx ，縱坐標為 dfn[x]dfn[x]，權值為這個數字。

• 那么詢問就相當于是詢問一個矩形的點權和。

• 為了支持平面上單點修改，矩形求和。

• 我們有兩種方法：

  1 . cdq分治+樹狀數組

• 我們把修改和詢問在一起按時間（題目中的 tt）排序，一共 n+m+qn+m+q 個東西。

• 我們從 [1,n+m+q][1,n+m+q] 開始遞歸；

• 對于遞歸一個 [l,r][l,r] 的區間，設 mid=(l+r)/2mid=(l+r)/2

• 我們只考慮 [l,mid][l,mid] 的修改對 [mid+1,r][mid+1,r] 的詢問的貢獻；

• 這之后，遞歸到 [l,mid][l,mid] , [mid+1,r][mid+1,r] 中繼續處理。

• 現在我們就只用考慮 [l,mid][l,mid] 的修改對 [mid+1,r][mid+1,r] 的詢問的貢獻；

• 我們可以簡單地對 [l,mid][l,mid] 中的修改的橫坐標排序，然后把縱坐標插入樹狀數組中，

• 那么 [mid+1,r][mid+1,r] 中的詢問就能通過查詢樹狀數組來實現統計的過程。

  2 . 二維數據結構

• 這個就直接做就好了。

• 只要你會任何二維的數據結構就可以。

• 推薦學習：樹狀數組套線段樹。

• 時間復雜度 O(Nlog2N)O(Nlog2N) 。

Code

#include<cstdio> #include<cstring> #include<algorithm> #include<cctype> using namespace std; const int N=1e5+5; struct data {int ty,t,x,w; }a[N*3]; int tot,num; int first[N],nex[N<<1],en[N<<1]; int dep[N],dfn[N],size[N]; int id[N*3]; long long f[N],ans[N]; inline int read() {int X=0,w=0; char ch=0;while(!isdigit(ch)) w|=ch=='-',ch=getchar();while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();return w?-X:X; } inline void insert(int x,int y) {nex[++tot]=first[x];first[x]=tot;en[tot]=y; } void dfs(int x,int y) {dfn[x]=++tot;dep[x]=dep[y]+1;size[x]=1;for(int i=first[x];i;i=nex[i])if(en[i]^y){dfs(en[i],x);size[x]+=size[en[i]];} } inline bool cmpt(data x,data y) {return x.t<y.t || x.t==y.t && !x.ty && y.ty; } inline bool cmpx(int x,int y) {return a[x].t>a[y].t; } inline void change(int x,int y) {while(x<N) f[x]=y?f[x]+y:0,x+=x&-x; } inline long long find(int x) {long long sum=0;while(x) sum+=f[x],x-=x&-x;return sum; } void solve(int l,int r) {if(l==r) return;int mid=l+r>>1;int nn=l-1,mm=mid;for(int i=l;i<=mid;i++)if(!a[i].ty) id[++nn]=i;for(int i=mid+1;i<=r;i++)if(a[i].ty) id[++mm]=i;if(nn>=l && mm>mid){sort(id+l,id+1+nn,cmpx);sort(id+mid+1,id+1+mm,cmpx);for(int i=mid+1,j=l;i<=mm;i++){while(j<=nn && a[id[j]].t>=a[id[i]].t){change(dfn[a[id[j]].x],a[id[j]].w);j++;}ans[a[id[i]].w]+=find(dfn[a[id[i]].x]+size[a[id[i]].x]-1)-find(dfn[a[id[i]].x]-1);}for(int i=l;i<=nn;i++) change(dfn[a[id[i]].x],0);}if(nn>=l && nn<mid) solve(l,mid);if(mm>mid && mm<r) solve(mid+1,r); } int main() {freopen("appletree.in","r",stdin);freopen("appletree.out","w",stdout);int n=read(),m=read(),q=read();for(int i=1;i<=n;i++){a[++num].x=i;a[num].t=1;a[num].w=read();}for(int i=1;i<n;i++){int x=read(),y=read();insert(x,y);insert(y,x);}dfs(1,tot=0);for(int i=1;i<=m;i++){a[++num].t=read()+1;a[num].x=read();a[num].w=read();}for(int i=1;i<=q;i++){a[++num].ty=1;a[num].t=read();a[num].x=read();a[num].w=i;}sort(a+1,a+1+num,cmpt);for(int i=1;i<=num;i++) a[i].t+=dep[a[i].x];solve(1,num);for(int i=1;i<=q;i++) printf("%lld\n",ans[i]);return 0; }

總結

以上是生活随笔為你收集整理的JZOJ 5699. 【gdoi2018 day1】涛涛接苹果（appletree）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。