【JZOJ 省选模拟】6703.tree

题目

Description

Input

Output
输出到文件tree.out 。
若干行，每行一个数表示对应询问的答案。

Sample Input
5 13
1 2 3 4 5
1 2
2 3
3 4
2 5
2 1
2 2
2 3
2 4
2 5
2 6
1 2 6
2 1
2 2
2 3
2 4
2 5
2 6

Sample Output
1
1
2
2
1
0
1
1
1
2
1
1

Data Constraint
本题采用捆绑测试，必须通过子任务下所有测试点方可得分。
子任务1（10分）：n,q<=5000 。
子任务2（20分）：保证每个点度数小于等于20。
子任务3（10分）：整棵树以1为根，其他点都连向1。
子任务4（25分）： n,q<=30000 。
子任务5（35分）：没有任何限制。
对于所有数据，n，q<=5*10^{5，所有权值任何时候均在[1,10}6]，保证操作1 中所有 单调递增。

温馨提示，注意读入效率，输入文件大小大约16M。

思路

这题简单版数据太水了，我一个错误性显然的AC了
首先答案等于满足的点-两端满足的边
考虑子任务2和3
2的话出度只有20，所以一次修改只涉及20个点
所以对于一个点，暴力判断它的出边，然后用树状数组维护一下

考虑菊花图
发现我们只需要关心比点1大的点，用一个set维护就行

那么，我们合并这两种做法
每次对于一个点维护这个点连边中另一端比这个点权值大的出边，每次暴力遍历这样的出边并看是否需要改方向，并在树状数组上修改，由于权值不降，复杂度也是均摊 O(1)。

代码

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+77,inf=1e6+77;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (!isdigit(ch)) {if (ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int tr[N*2],n,ls[N],fa[N],top,a[N],q,d[N],cnt;
struct E
{
	int to,next;
}e[N*2];
multiset<int> g[N];
void add(int u,int v)
{
	e[++cnt].to=v; e[cnt].next=ls[u]; ls[u]=cnt;
}
void ins(int x,int c)
{
	for(int i=1; i<=inf; i+=i&-i) tr[i]+=c;
	for(int i=++x; i<=inf; i+=i&-i) tr[i]-=c;
}
int query(int x)
{
	int yjy=0;
	for(int i=x; i; i-=i&-i) yjy+=tr[i];
	return yjy;
}
void init(int x)
{
	queue<int> q;
	q.push(x);
	while(!q.empty())
	{
		int u=q.front(); q.pop();
		ins(a[u],1);
		for(int i=ls[u]; i; i=e[i].next)
		{
			int v=e[i].to;
			if(v==fa[u])continue;
			fa[v]=u; q.push(v);
			ins(min(a[u],a[v]),-1);
			g[u].insert(a[v]);
			d[u]+=(a[v]>=a[u]);
		}
	}
}
int main()
{
	freopen("tree.in","r",stdin);freopen("tree.out","w",stdout);
	scanf("%d%d",&n,&q);
	for(int i=1; i<=n; i++) a[i]=read();
	for(int i=1; i<n; i++)
	{
		int x=read(),y=read();
		add(x,y);add(y,x);
	}
	init(1);
	while(q--)
	{
		int op=read(),x,v;
		if(op==2)
		{
			x=read();
			printf("%d\n",query(x));
		}
		else
		{
			x=read(); v=read();
			if(fa[x])
			{
				ins(min(a[fa[x]],a[x]),1);
				ins(min(a[fa[x]],v),-1);
				g[fa[x]].erase(g[fa[x]].lower_bound(a[x]));
				g[fa[x]].insert(v);
				if(a[x]<a[fa[x]]&&v>=a[fa[x]]) d[fa[x]]++;
			}
			multiset<int> ::iterator it;
			if(g[x].begin()!=g[x].end()&&*--g[x].end()>=a[x])
			{
				it=g[x].lower_bound(a[x]);
				int yjy=0;
				for(; it!=g[x].end()&&*it<v; it++) ins(*it,-1),++yjy;
				ins(a[x],d[x]);
				d[x]-=yjy;
				ins(v,-d[x]);
			}
			ins(a[x],-1);
			ins(v,1);
			a[x]=v;
		}
	}
}