Description
给出n个数,以此插入一棵平衡二叉树中,保证一个节点的左儿子要小于该点,右儿子要大于该点,问在插入过程中需要多少次比较
Input
第一行一整数n表示数的个数,之后是一个1~n的排列(1<=n<=200000)
Output
输出在构成平衡二叉树的过程中需要比较的次数
Sample Input
3
2 1 3
Sample Output
2
Solution
开一个set记录当前已经插入二叉树中的数,开一个数组记录每个数在二叉树中的深度(即为插入这个数需要比较的次数),对于当前插入的数,找到set中它前面和后面的数表示它可能要插入的地方,两边谁深度深肯定要插在谁下面(因为这个数会提前和深度浅的比较后往下接着找到深度深的),那么深度深的点的深度加一就是当前点的深度,也就是这个点插入需要的比较次数
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 2222222
int n,p,a[maxn];
set<int>s;
set<int>::iterator it;
ll ans=0;
int main()
{
while(~scanf("%d",&n))
{
s.clear();
ans=0;
memset(a,0,sizeof(a));
for(int i=1;i<=n;i++)
{
scanf("%d",&p);
int temp=-1;
it=s.lower_bound(p);
if(it!=s.end())temp=max(temp,a[*it]);
if(it!=s.begin())it--,temp=max(temp,a[*it]);
temp++;
ans+=temp;
a[p]=temp;
s.insert(p);
}
printf("%I64d\n",ans);
}
return 0;
}