傳送門
文章目錄
題意：
思路：
第一次聽說樹鏈剖分能在fa[top[i]]fa[top[i]]fa[top[i]]的地方加懶標記，學到了學到了。
首先不能被題目嚇住，這個題目仔細剖析一下不難發現一些性質：
以下用seise_isei?表示iii子樹的大小。
(1)(1)(1)當uuu選擇了fff點，那么vvv選任一點都可以，也就是讓fff為根的子樹的每個點權值都加1sef1sefk\frac{1}{se_f}\frac{1}{se_f}ksef?1?sef?1?k。
(2)(2)(2)當uuu選擇了非fff點的時候，其選擇的點假設落在與fff直接相連的xxx點所在的子樹中，那么vvv只能選非xxx的子樹中的點，也就是讓非xxx的子樹的點都加上sexsef1sef?sexk\frac{se_x}{se_f}\frac{1}{se_f-se_x}ksef?sex??sef??sex?1?k。
以上兩個操作顯然可以先將fff所在的子樹都加上1sef1sefk+∑sexsef1sef?sexk\frac{1}{se_f}\frac{1}{se_f}k+\sum \frac{se_x}{se_f}\frac{1}{se_f-se_x}ksef?1?sef?1?k+∑sef?sex??sef??sex?1?k，現在就有個比較顯然的做法就是遍歷fff的出邊給子樹減去sexsef1sef?sexk\frac{se_x}{se_f}\frac{1}{se_f-se_x}ksef?sex??sef??sex?1?k，fff的全部子樹都加上1sef1sefk+∑sexsef1sef?sexk\frac{1}{se_f}\frac{1}{se_f}k+\sum \frac{se_x}{se_f}\frac{1}{se_f-se_x}ksef?1?sef?1?k+∑sef?sex??sef??sex?1?k即可。
但這樣的話給你個菊花圖就炸了，復雜度升到qnlognqnlognqnlogn，由于是對子樹操作的，比較容易想到能不能加個懶標記呢？我們加了懶標記又如何在計算答案的時候能減去這個懶標記呢？解決了這個兩個問題我們就可以順利解決這個問題了。
樹上倍增解決這個題不是很容易，我們考慮重鏈剖分。
首先我們加點的時候，思路就是按照上面哪個思路，即先給所有點都加上，再給子樹減去某個值，所以我們lazylazylazy標記應該是標記了減去了多少。首先加操作顯然可以寫一個dfsdfsdfs序，讓后用樹狀數組維護一下。考慮書剖的特殊性質，我們只需要在tag[f]tag[f]tag[f]加上kkk，讓后將重兒子直接減去sexsef1sef?sexk\frac{se_x}{se_f}\frac{1}{se_f-se_x}ksef?sex??sef??sex?1?k，因為重兒子在之后跳toptoptop的時候會直接跳過的，而輕兒子在跳到這條鏈的top[i]top[i]top[i]的時候，需要減去s2[top[i]]?tag[fa[top[i]]]s2[top[i]]*tag[fa[top[i]]]s2[top[i]]?tag[fa[top[i]]]，s2s2s2是預處理的sexsef1sef?sex\frac{se_x}{se_f}\frac{1}{se_f-se_x}sef?sex??sef??sex?1?。
這樣問題就順利解決啦~
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
#define lowbit(x) ((x)&(-x))
using namespace std
;
typedef long long LL
;
typedef unsigned long long ULL
;
typedef pair
<int,int> PII
;const int N
=200010,mod
=998244353,INF
=0x3f3f3f3f;
const double eps
=1e-6;int n
,m
;
vector
<int>v
[N
];
int se
[N
],son
[N
],top
[N
],fa
[N
],in
[N
],tot
;
LL inv
[N
],s1
[N
],s2
[N
],s3
[N
];
LL tr
[N
],tag
[N
];void dfs1(int u
,int f
) {se
[u
]=1; fa
[u
]=f
;for(auto x
:v
[u
]) {if(x
==f
) continue;dfs1(x
,u
);se
[u
]+=se
[x
];if(se
[son
[u
]]<se
[x
]) son
[u
]=x
;}s1
[u
]=inv
[se
[u
]]*inv
[se
[u
]]%mod
;for(auto x
:v
[u
]) {if(x
==f
) continue;s2
[x
]=se
[x
]*inv
[se
[u
]]%mod
*inv
[se
[u
]-se
[x
]]%mod
;s3
[u
]+=s2
[x
]; s3
[u
]%=mod
;}
}void dfs2(int u
,int t
) {top
[u
]=t
; in
[u
]=++tot
;if(son
[u
]) dfs2(son
[u
],t
); for(auto x
:v
[u
]) {if(x
==fa
[u
]||x
==son
[u
]) continue;dfs2(x
,x
);}
}void add(int x
,LL c
) {for(int i
=x
;i
<=n
;i
+=lowbit(i
)) tr
[i
]+=c
,tr
[i
]%=mod
,tr
[i
]+=mod
,tr
[i
]%=mod
;
}LL
query(int x
) {LL ans
=0;for(int i
=x
;i
;i
-=lowbit(i
)) ans
+=tr
[i
],ans
%=mod
;return ans
;
}int main()
{
inv
[1]=1;for(int i
=2;i
<N
;i
++)inv
[i
]=(mod
-mod
/i
)*inv
[mod
%i
]%mod
;int _
; scanf("%d",&_
);while(_
--) {scanf("%d%d",&n
,&m
); tot
=0;for(int i
=1;i
<=n
;i
++) se
[i
]=son
[i
]=top
[i
]=fa
[i
]=s3
[i
]=s2
[i
]=s1
[i
]=0,v
[i
].clear();for(int i
=1;i
<=n
-1;i
++) {int a
,b
; scanf("%d%d",&a
,&b
);v
[a
].pb(b
); v
[b
].pb(a
);}dfs1(1,0); dfs2(1,1);while(m
--) {int op
,x
; scanf("%d%d",&op
,&x
);if(op
==1) {int k
; scanf("%d",&k
);add(in
[x
],1ll*(s1
[x
]+s3
[x
])%mod
*k
%mod
); add(in
[x
]+se
[x
],(((mod
-s1
[x
]-s3
[x
])%mod
+mod
)%mod
)*k
%mod
);if(son
[x
]) add(in
[son
[x
]],(mod
-s2
[son
[x
]]*k
%mod
)%mod
),add(in
[son
[x
]]+se
[son
[x
]],s2
[son
[x
]]*k
%mod
);tag
[x
]+=k
; tag
[x
]%=mod
;} else {LL ans
=query(in
[x
]);for(int i
=top
[x
];fa
[i
];i
=top
[fa
[i
]]) {ans
-=tag
[fa
[i
]]*s2
[i
]%mod
,ans
+=mod
,ans
%=mod
;}printf("%lld\n",ans
%mod
);}}for(int i
=1;i
<=n
;i
++) tr
[i
]=tag
[i
]=0;}return 0;
}
創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎
總結
以上是生活随笔為你收集整理的兰州大学第一届 飞马杯 ★★快乐苹果树★★ 树链剖分 + 懒标记 + 树状数组的全部內容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。
