暴力 【p4092】[HEOI2016/TJOI2016]树

Description

在2016年，佳媛姐姐刚刚学习了树，非常开心。现在他想解决这样一个问题：给定一颗有根树（根为1），有以下两种操作：

1. 标记操作：对某个结点打上标记（在最开始，只有结点1有标记，其他结点均无标记，而且对于某个结点，可以打多次标记。）
2. 询问操作：询问某个结点最近的一个打了标记的祖先（这个结点本身也算自己的祖先）

你能帮帮他吗?

Input

输入第一行两个正整数N和Q分别表示节点个数和操作次数

接下来N-1行，每行两个正整数u,v(1≤u,v≤n)表示u到v有一条有向边

接下来Q行，形如“opernum”oper为“C”时表示这是一个标记操作,oper为“Q”时表示这是一个询问操作对于每次询问操作。

Output

输出一个正整数，表示结果。

令人窒息的一个题,洛谷上暴力(A)了？

但是bzoj上T的飞起 emm

听学长说,这题是树剖。

但是暴力(A)掉了。于是写正解就向后拖一拖 emmm

小优化：如果没有修改操作,直接输出(1)。

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#define R register

using namespace std;

const int gz=100008;

inline void in(int &x)
{
	int f=1;x=0;char s=getchar();
	while(!isdigit(s)){if(s=='-')f=-1;s=getchar();}
	while(isdigit(s)){x=x*10+s-'0';s=getchar();}
	x*=f;
}

int n,q;

int f[gz],tg[gz];


char opt[5];

inline int work(R int x)
{
	if(tg[x])return x;
	while(f[x])
	{
		x=f[x];
		if(tg[x])return x;
	}
}

bool flg;

int main()
{
	in(n),in(q);
	for(R int i=1,x,y;i<n;i++)
		in(x),in(y),f[y]=x;
	tg[1]=true;
	for(R int i=1,x;i<=q;i++)
	{
		scanf("%s",opt+1);
		if(opt[1]=='Q')
		{
			in(x);
			if(!flg)puts("1");
			else printf("%d
",work(x));
		}
		else
		{
			in(x),flg=true;
			tg[x]=true;
		}
	}
}