线段树合并 || BZOJ 5457: 城市
题面：https://www.lydsy.com/JudgeOnline/problem.php?id=5457
题解：
线段树合并，对于每个节点维护sum（以该节点为根的子树中最大的种类和）和kind（以该节点为根的子树中种类和最大的种类）即可。
代码：
1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 #include<algorithm> 5 using namespace std; 6 const int maxn=(4e5)+50; 7 int N,M,U,V,num_edge=0,edge_head[maxn],root[maxn]; 8 int num_treenode=0; 9 struct Edge{int to,nx;}edge[maxn<<1]; 10 inline void Add_edge(int from,int to){ 11 edge[++num_edge].nx=edge_head[from]; 12 edge[num_edge].to=to; 13 edge_head[from]=num_edge; 14 return; 15 } 16 struct Tree{int lc,rc,l,r,sum,kd;}tr[maxn*20]; 17 inline void Pushup(int x){ 18 int lc=tr[x].lc,rc=tr[x].rc; 19 if(tr[lc].sum>=tr[rc].sum){ 20 tr[x].sum=tr[lc].sum; 21 tr[x].kd=tr[lc].kd; 22 } 23 else { 24 tr[x].sum=tr[rc].sum; 25 tr[x].kd=tr[rc].kd; 26 } 27 return; 28 } 29 inline void Build(int x,int l,int r,int q,int s){ 30 tr[x].l=l;tr[x].r=r;int mid=(l+r)>>1; 31 if(l==r&&l==q){ 32 tr[x].kd=q; 33 tr[x].sum=s; 34 return; 35 } 36 if(q<=mid)Build(tr[x].lc=++num_treenode,l,mid,q,s); 37 else Build(tr[x].rc=++num_treenode,mid+1,r,q,s); 38 Pushup(x); 39 return; 40 } 41 struct A_{int kd,sum;}A[maxn]; 42 inline int Merge(int u,int v){ 43 if(!u)return v; 44 if(!v)return u; 45 int l=tr[u].l,r=tr[u].r; 46 if(l==r){ 47 tr[u].sum+=tr[v].sum; 48 return u; 49 } 50 tr[u].lc=Merge(tr[u].lc,tr[v].lc); 51 tr[u].rc=Merge(tr[u].rc,tr[v].rc); 52 Pushup(u); 53 return u; 54 } 55 inline void Dfs(int x,int fa){ 56 for(int i=edge_head[x];i;i=edge[i].nx){ 57 int y=edge[i].to; 58 if(y!=fa){ 59 Dfs(y,x); 60 Merge(root[x],root[y]); 61 } 62 } 63 return; 64 } 65 int main(){ 66 scanf("%d%d",&N,&M); 67 for(int i=1;i<N;i++){ 68 scanf("%d%d",&U,&V); 69 Add_edge(U,V);Add_edge(V,U); 70 } 71 for(int i=1;i<=N;i++){ 72 scanf("%d%d",&A[i].kd,&A[i].sum); 73 Build(root[i]=++num_treenode,1,M,A[i].kd,A[i].sum); 74 } 75 Dfs(1,0); 76 for(int i=1;i<=N;i++)printf("%d %d\n",tr[root[i]].kd,tr[root[i]].sum); 77 return 0; 78 }
By:AlenaNuna
