GYM 100090 F.Asperger Syndrome（set）

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;
}