萌新求助线段树合并
查看原帖
萌新求助线段树合并
171487
cmll02楼主2021/8/30 20:50

刚开始写一遍 WA 12。

然后发现这题线段树合并的时候开新节点。

那我每次都复制一份节点出来,按理说空间只是两倍,但是我写完一遍之后瞬间 MLE 了qq_emoji: ll

// 从未在意的名字永远不会被提起 雪葉/鹤见江野

/*
+
++
+++
++++
+++++
++++++
+++++++
++++++++
+++++++++
++++++++++
+++++++++++
++++++++++++
+++++++++++++
++++++++++++++
+++++++++++++++
++++++++++++++++
+++++++++++++++++
++++++++++++++++++
+ +++++++++++++++++
+  ++++++++++++++ ++
+   +++++++++++++  ++
+    ++++++++++++   ++
+     +++++++++++    ++
+      ++++++++++     ++
+       +++++++++      ++
+        ++++++++       ++
+         +++++++++++++++++
+          +++++++++++++++++
+           ++++++++++++++
+            +++++++++++
+             ++++++++
+              +++++
+               ++
+               +
+               +
+              ++
+             +++
+            ++++
+           +++++
+           +++++
+           +++++
+           +++++
+     +     +++++
+    +++    +++++
+   ++ ++   +++++
+  ++   ++  +++++
+ ++  +  ++ +++++
+++  +++  +++++++
++  ++ ++  ++++++
 
 
 ++++++++      +++++++++++     +++      +++        ++++++++        ++++++++
+++++++++     +++++++++++++    +++      +++       +++    +++      +++    +++
+++          +++   +++   +++   +++      +++      +++   ++++++    +++      +++
+++          +++   +++   +++   +++      +++      +++ +++  +++           +++
+++          +++   +++   +++   +++ ++   +++ ++   ++++++   +++         +++
+++++++++    +++   +++   +++   +++ ++   +++ ++    +++    +++        +++    ++
 ++++++++    +++   +++   +++   +++++    +++++      ++++++++       +++++++++++
*/
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <set>
#define nullptr NULL
#define od(x) printf("%d",x)
#define odb(x) printf("%d ",x)
#define odl(x) printf("%d\n",x)
#define odp(x,y) printf("%d %d\n",x,y)
#define ol(x) puts("")
#define old(x) printf("%lld",x)
#define oldb(x) printf("%lld ",x)
#define oldl(x) printf("%lld\n",x)
#define oldp(x,y) printf("%lld %lld\n",x,y)
#define rg(x) for(int i=1;i<=(x);i++){
#define rg_(i,x) for(int i=1;i<=(x);i++){
#define gr }
#define rrg(x) for(int i=0;i<(x);i++)
#define rdln(a) a[i]=read();
#define rdln0(a,x) rrg(x) rdln(a) gr
#define rdln1(a,x) rg(x) rdln(a) gr
//#define int long long
#define newe(n) struct Edge{int v,w,nxt;}e[n*2+5];\
typedef int arr[n+5];\
arr h;\
int cnt=1;\
inline void addedge(int u,int v,int w){e[cnt]=(Edge){v,w,h[u]};h[u]=cnt++;}
#define mgs int fa[1<<22],sz[1<<22],t[1<<22];\
inline int f(int x){return x==fa[x]?x:fa[x]=f(fa[x]);}\
inline int u(int x,int y)\
{\
    int fx=f(x),fy=f(y);\
    if(fx==fy)return 0;\
    if(sz[fx]>sz[fy])fx^=fy^=fx^=fy;\
    fa[fx]=fy,sz[fy]+=sz[fx];\
    return 1;\
}
inline int read()
{
    int num=0,f=1;char c=getchar();
    while(c<48||c>57){if(c=='-')f=-1;c=getchar();}
    while(c>47&&c<58)num=num*10+(c^48),c=getchar();
    return num*f;
}
inline int re1d()
{
    char c=getchar();
    while(c<48||c>49)c=getchar();
    return c&1;
}
inline int min(int a,int b){return a>b?b:a;}
inline int max(int a,int b){return a<b?b:a;}
struct Node
{
	Node *lc,*rc;
	int sum;
	Node(){lc=rc=NULL,sum=0;}
	Node(int x):sum(x){}
}*root[300005],*rot[300005];
inline void maintain(Node *&o,int l,int r)
{
	if(l==r)return;
	o->sum=0;
	if(o->lc)o->sum+=o->lc->sum;
	if(o->rc)o->sum+=o->rc->sum;
}
Node* merge(Node *&a,Node *b,int l,int r)
{
	if(a==nullptr)return a=b;
	if(b==nullptr)return a;
	if(l==r)
	{
		a->sum+=b->sum;
		return a;
	}
	int m=l+r>>1;
	a->lc=merge(a->lc,b->lc,l,m);
	a->rc=merge(a->rc,b->rc,m+1,r);
	maintain(a,l,r);
	return a;
}
void insert(Node *&o,int l,int r,int p,int v)
{
	if(o==nullptr)o=new Node();
	if(l==r)
	{
		o->sum+=v;
		return;
	}
	int m=l+r>>1;
	if(p<=m)insert(o->lc,l,m,p,v);
	else insert(o->rc,m+1,r,p,v);
	maintain(o,l,r);
}
newe(300005);
int pigstd=100000;
int dep[300005],ans[300005],sz[300005];
void dfs(int u,int fa)
{
	dep[u]=dep[fa]+1;sz[u]=1;
	for(int i=h[u];i;i=e[i].nxt)if(e[i].v!=fa)dfs(e[i].v,u),sz[u]+=sz[e[i].v];
}
void cpy(Node *&o,Node *p)
{
	if(p==nullptr)return;
	if(o==nullptr)o=new Node();
	o->sum=p->sum;
	cpy(o->lc,p->lc);
	cpy(o->rc,p->rc);
}
void pigstd_AK_IOI(int u,int fa)
{
	for(int i=h[u];i;i=e[i].nxt)
	{
		int v=e[i].v;
		if(v==fa)continue;
		pigstd_AK_IOI(v,u);
		merge(root[u],root[v],1,pigstd);
	}
	cpy(rot[u],root[u]);
}
int query(Node *o,int l,int r,int L,int R)
{
	if(o==nullptr)return 0;
	if(L<=l&&r<=R)return o->sum;
	int m=l+r>>1;
	int s=0;
	if(L<=m)s+=query(o->lc,l,m,L,R);
	if(m<R)s+=query(o->rc,m+1,r,L,R);
	return s;
}
signed main()
{
	int n=read(),m=read();pigstd=n;
	rg(n-1)int x=read(),y=read();addedge(x,y,1),addedge(y,x,1);gr
	dfs(1,0);
	rg(n)insert(root[i],1,n,dep[i],sz[i]-1);gr
	pigstd_AK_IOI(1,1);
	rg(m)
	int a=read(),k=read();
	int ans=min(dep[a]-1,k)*(sz[a]-1);
	ans+=query(rot[a],1,n,dep[a]+1,dep[a]+k);
	odl(ans);
	gr
	return 0;
}
2021/8/30 20:50
加载中...