愛之箭發射（las）

目描述

?

小海是弓道部的成員，非常擅長射箭(Love Arrow Shoot)。今天弓道部的練習是要射一棵樹。一棵樹是一個nn個點n?1n?1條邊的無向圖，且這棵樹的第ii個點有一個值wiwi,wi∈[1,m]wi∈[1,m]。每一次小海會射中樹的一條邊，并將這條邊移除。此外，小海定義一棵樹的las值為∑vi?i∑vi?i，vivi為這棵樹中第ii小的wiwi。現在小海會告訴你她射中的邊的順序，你需要回答每一次她射中的邊所在的樹的las值，之后被射中的邊會被移除。答案mod998244353mod998244353

?

輸入

?

第一行兩個數n,mn,m

第二行nn個數wiwi

接下來n?1n?1行每行兩個數ai,biai,bi,表示初始的樹第ii條邊連接aiai和bibi。

接下來n?1n?1行每行一個數表示射中的邊。

?

?

輸出

?

n-1行每行一個數表示射中的邊的樹的las值

?

?

樣例輸入

5 4396 2 3 1 4 5 1 2 1 3 2 4 2 5 4 1 2 3

樣例輸出

55 30 5 11

提示

?

數據范圍：

前20% n<=1e3

另外20% m<=10

另外20%?保證第i條邊連接i和i+1

另外20% n<=1e5

100% n<=5e5 wi<=m<=1e4

下發一個樣例滿足第一個部分分。

?

?

來源

noip2018模擬-robinliu

---

solution

把操作倒過來，看成加邊。

對于每個連通塊可以用一棵值域線段樹維護每個值出現的次數，值的和，還有la值。

合并時，新的la值可以由值的和*之前的值出現的次數得到。

比如 1*1+2*2+3*3? +2*(1+2+3) -> 3*1+4*2+5*3

線段樹合并維護即可

注意x,y均沒有左右兒子的特殊情況

#include<cstdio> #include<iostream> #include<cstdlib> #include<cstring> #include<algorithm> #include<cmath> #define maxn 500005 #define mod 998244353 #define ll long long using namespace std; int n,m,st[maxn],ed[maxn],w[maxn],id[maxn]; int fa[maxn],root[maxn*18],tot; int ls[maxn*18],rs[maxn*18],num[maxn*18]; ll ans[maxn],sum[maxn*18],s[maxn*18]; void build(int &k,int l,int r,int pl){if(!k)k=++tot;if(l==r){num[k]=1;sum[k]=s[k]=pl;return;}int mid=l+r>>1;if(pl<=mid)build(ls[k],l,mid,pl);else build(rs[k],mid+1,r,pl);num[k]=num[ls[k]]+num[rs[k]];sum[k]=sum[ls[k]]+sum[rs[k]];s[k]=s[ls[k]]+s[rs[k]]; } int getf(int k){if(fa[k]==k)return k;fa[k]=getf(fa[k]);return fa[k]; } int merge(int x,int y,int la){if(!x||!y){return x+y;}ls[x]=merge(ls[x],ls[y],la);rs[x]=merge(rs[x],rs[y],la+num[ls[x]]);//printf("x:%d y:%d sumx:%d sumy:%d la:%d\n ls:%d %d %d rs:%d %d %d\n",x,y,sum[x],sum[y],la,sum[ls[x]],num[ls[x]],s[ls[x]],sum[rs[x]],num[rs[x]],s[rs[x]]);if(!ls[x]&&!rs[x]){sum[x]=sum[x]+sum[y]+(num[x]*s[y])%mod;sum[x]%=mod;num[x]=num[x]+num[y];s[x]=s[x]+s[y];return x;}sum[x]=sum[ls[x]]+sum[rs[x]]+(num[ls[x]]*s[rs[x]])%mod;sum[x]%=mod;num[x]=num[ls[x]]+num[rs[x]];s[x]=s[ls[x]]+s[rs[x]];//cout<<sum[x]<<' '<<num[x]<<endl;return x; } int main(){cin>>n>>m;for(int i=1;i<=n;i++){scanf("%d",&w[i]);fa[i]=i;build(root[i],1,m,w[i]);}for(int i=1;i<n;i++){scanf("%d%d",&st[i],&ed[i]);}for(int i=1;i<n;i++)scanf("%d",&id[i]);for(int i=n-1;i>=1;i--){int x=getf(st[id[i]]),y=getf(ed[id[i]]);//cout<<"---------\n";//cout<<x<<' '<<y<<endl;root[x]=merge(root[x],root[y],0);ans[i]=sum[root[x]];fa[y]=x;}for(int i=1;i<n;i++)printf("%lld\n",ans[i]);return 0; } /* 5 5 2 2 1 4 5 1 2 1 3 2 4 2 5 4 1 2 3 */

?

轉載于:https://www.cnblogs.com/liankewei/p/10358822.html

總結

以上是生活随笔為你收集整理的爱之箭发射（las）的全部內容，希望文章能夠幫你解決所遇到的問題。

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