C. Remove Extra One
time limit per test 2 seconds
memory limit per test 256 megabytes
You are given apermutation p of length n. Remove one element from permutation to make the number of records themaximum possible.
We remind thatin a sequence of numbers a1, a2, ..., ak the element ai is a record if for every integer j (1 ≤ j < i) the following holds: aj < ai.
Input
The first linecontains the only integer n (1 ≤ n ≤ 105) — the length of the permutation.
The second linecontains n integers p1, p2, ..., pn (1 ≤ pi ≤ n) — the permutation. All the integers are distinct.
Output
Print the only integer — theelement that should be removed to make the number of records the maximumpossible. If there are multiple such elements, print the smallest one.
Examples
Input
1
1
Output
1
Input
5
5 1 2 3 4
Output
5
Note
In the first example the only elementcan be removed.
【题意】
给定n以及一个1~n的全排列,如果一个数的左侧所有数都要比它小,那么这个数为关键数。
现在问在数列中删掉哪个数能使得剩下的数列中关键数最多,若有多种情况,输出最小的。
【思路】
其实就是考虑每一个数对关键数的影响。开始实现复杂了,后来才知道可以用树状数组轻松解决。
我们用num[i]来表示把i这个数删去后,关键数个数的变化值。
每次读入一个数,先查询在他前面的数的个数,即比它小的个数,如果等于i-1,说明前面的数都要比这个数小,那么如果删去这个数后,显然关键数个数减少了1.
如果等于i-2,说明前面只有一个比它大,那么只要删去前面那个最大的数,当前这个数就变成了关键数,关键数的个数加1.
处理完之后把当前值更新到树状数组中。
最后扫一遍即可。
#include <cstdio>
#include <cmath>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
#define mst(a,b) memset((a),(b),sizeof(a))
#define rush() int T;scanf("%d",&T);while(T--)
typedef long long ll;
const int maxn = 1000005;
const ll mod = 1e9+7;
const ll INF = 0x3f3f3f3f;
const double eps = 1e-9;
int tree[maxn],num[maxn];
int lowbit(int x)
{
return x&(-x);
}
void add(int pos)
{
while(pos<maxn)
{
tree[pos]++;
pos+=lowbit(pos);
}
}
int query(int pos)
{
int ans=0;
while(pos>0)
{
ans+=tree[pos];
pos-=lowbit(pos);
}
return ans;
}
int main()
{
int n,x;
scanf("%d",&n);
int Max=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&x);
Max=max(Max,x);
int cnt=query(x);
if(cnt==i-1) num[Max]--;
else if(cnt==i-2) num[Max]++;
add(x);
}
int res=-1,ans=1;
for(int i=1;i<=n;i++)
{
if(num[i]>res)
{
res=num[i];
ans=i;
}
}
printf("%d\n",ans);
}