[CSP-S模拟测试]:树(树上上升序列+主席树+线段树)

题目传送门(内部题78)


输入格式

  第一行输入两个整数$n,q$,表示节点数和询问数。
  第二行输入$n$个整数$w_i$,表示第$i$个点的智商。
  第三行至第$n+1$行每行输入两个数$x,y$,表示树上一条边。
  第$n+2$行至第$n+q+1$行每行三个数$u,v,c$表示一次探究。(保证$v$是$u$的祖先)


输出格式

  输出$q$行,每行两个数表示探究过程中$cwystc$需要努力学习的次数。


样例

见下发文件


数据范围与提示

  对于$10\%$的数据:$n\leqslant 1,000$
  对于另外$30\%$的数据:家谱树为一条链
  对于$100\%$的数据:$n,q,w_i,c\leqslant 100,000$


题解

转化一下题意,就是让我们求从$u$到$v$的上升序列长度(注意这里的上升序列是指碰见比它大的就选中,而不是最长上升子序列)。

剩下就是码力问题了……

我可能打的比较麻烦,用了主席树和线段树优化……

时间复杂度:$\Theta(n\log n)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
#define L(x) x<<1
#define R(x) x<<1|1
using namespace std;
struct rec{int nxt,to;}e[200000];
int head[100001],cnt;
int n,q;
int w[100001];
int son[100001],size[100001],top[100001],fa[100001],dfn[100001],rk[100001],tim,root[5000001],lson[5000001],rson[5000001],num[5000001],tot;
int tr[400001];
void add(int x,int y)
{
	e[++cnt].nxt=head[x];
	e[cnt].to=y;
	head[x]=cnt;
}
void pushups(int x){num[x]=num[lson[x]]+num[rson[x]];}
void pushupm(int x){tr[x]=max(tr[L(x)],tr[R(x)]);}
void build(int x,int l,int r)
{
	if(l==r){tr[x]=w[rk[l]];return;}
	int mid=(l+r)>>1;
	build(L(x),l,mid);
	build(R(x),mid+1,r);
	pushupm(x);
}
void adds(int &x,int f,int l,int r,int w)
{
	if(!x)x=++tot;
	if(l==r){num[x]=1;return;}
	int mid=(l+r)>>1;
	if(w<=mid)
	{
		rson[x]=rson[f];
		adds(lson[x],lson[f],l,mid,w);
	}
	else adds(rson[x],rson[f],mid+1,r,w);
	pushups(x);
}
int asks(int x,int l,int r,int w)
{
	if(!x)return 0;
	if(w<=l)return num[x];
	int mid=(l+r)>>1;
	if(w<=mid)return asks(lson[x],l,mid,w)+num[rson[x]];
	else return asks(rson[x],mid+1,r,w);
}
int askm(int x,int l,int r,int L,int R)
{
	if(r<L||R<l)return 0;
	if(L<=l&&r<=R)return tr[x];
	int mid=(l+r)>>1;
	return max(askm(L(x),l,mid,L,R),askm(R(x),mid+1,r,L,R));
}
void dfs(int x,int f)
{
	size[x]=1;
	top[x]=x;
	adds(root[x],root[f],0,1000000,w[x]);
	for(int i=head[x];i;i=e[i].nxt)
	{	
		if(size[e[i].to])continue;
		fa[e[i].to]=x;
		dfs(e[i].to,x);
		size[x]+=size[e[i].to];
		if(size[son[x]]<size[e[i].to])son[x]=e[i].to;
	}
}
void dfs(int x)
{
	dfn[x]=++tim;
	rk[tim]=x;
	if(son[x])
	{
		top[son[x]]=top[x];
		dfs(son[x]);
	}
	for(int i=head[x];i;i=e[i].nxt)
		if(!dfn[e[i].to])dfs(e[i].to);
}
int query(int x,int y,int c)
{
	int now=top[x],mx=c;
	int res=asks(root[x],0,1000000,c+1);
	while(now!=top[y])
	{
		mx=max(mx,askm(1,1,n,dfn[now],dfn[x]));
		x=fa[now];
		now=top[x];
	}
	mx=max(mx,askm(1,1,n,dfn[y],dfn[x]));
	res-=asks(root[fa[y]],0,1000000,mx+1);
	return res;
}
int main()
{
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++)scanf("%d",&w[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v);add(v,u);
	}
	dfs(1,0);
	dfs(1);
	build(1,1,n);
	while(q--)
	{
		int u,v,c;
		scanf("%d%d%d",&u,&v,&c);
		printf("%d\n",query(u,v,c));
	}
	return 0;
}

rp++

posted @ 2019-10-22 19:27  HEOI-动动  阅读(275)  评论(0编辑  收藏  举报