Codeforces 797D Broken BST dfs+模拟

点击打开链接

题意:n<=1e5的树,最多有两个分支,要求按照排序二叉树的方式询问值是否存在,求出错的次数(即a[i]存在但是找不到)?
若暴力查询 最坏时,树退化成链O(n^2)  
因为是树 使用root->值x路径唯一,一个值x能被查询到,当且仅当从root起往值x走的路径中,端点p向右走的值都比x小,端点q向左走的值都比x大
则x>=max(p) && x<=min(q)  

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+20;
const int inf=1e9;
struct node{
	int l,r,v;
}t[N];
int n,root,cnt;
bool rt[N];
map<int,int> mp,vis;
void dfs(int u,int p,int q)// 
{
	if(u==-1)
		return;
	if(t[u].v>=p&&t[u].v<=q)
	{
		if(mp[t[u].v])
			mp[t[u].v]=0;// 
	}
	dfs(t[u].l,p,min(q,t[u].v));
	dfs(t[u].r,max(p,t[u].v),q);
}
int main()
{
	while(cin>>n)
	{
		memset(rt,true,sizeof(rt));
		mp.clear();
		int num=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d%d%d",&t[i].v,&t[i].l,&t[i].r);
			if(t[i].l!=-1)
				rt[t[i].l]=false;
			if(t[i].r!=-1)
				rt[t[i].r]=false;//rt无前驱 
			mp[t[i].v]++;
		}
		cnt=0; 
		for(int i=1;i<=n;i++)
			if(rt[i])
				root=i;
		dfs(root,-inf,inf);
		for(int i=1;i<=n;i++)
		{
			cnt+=mp[t[i].v];//总共进行n次 若值x无法询问 则答案加上mp[x]次 
			mp[t[i].v]=0;
		}
		cout<<cnt<<endl;
	}
	return 0;
}