C. Remove Extra One
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
You are given a permutation p of length n. Remove one element from permutation to make the number of records the maximum possible.
We remind that in 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 line contains the only integer n (1 ≤ n ≤ 105) — the length of the permutation.
The second line contains n integers p1, p2, ..., pn (1 ≤ pi ≤ n) — the permutation. All the integers are distinct.
Output
Print the only integer — the element that should be removed to make the number of records the maximum possible. 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 element can be removed.
题意:一个长度为n的排列,删除一个数使得rec数最多,rec数为这个数左边的数均比它小
思路:开始读错题了呀,以为rec数是左边单调递增的,诶诶。
计算每个数对rec数的贡献,即这个数是唯一比它右边数大的,然后贡献最多的就是,用BIT维护了一个当前数是否为rec数或rec备选数。
最后处理部分也搞得很蒙,写的乱七八糟。
A了后搜了一下,发现只要维护一个最大数和次大数就可以,果然,我就是个zz
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#define maxn 100005
#define INF 0x3f3f3f3f
typedef long long ll;
using namespace std;
int n, c[maxn];
int lowbit(int x)
{
return x & (-x);
}
void add(int k, int num)
{
while(k <= n){
c[k] += num;
k += lowbit(k);
}
}
int Sum(int k)
{
int sum = 0;
while(k){
sum += c[k];
k -= lowbit(k);
}
return sum;
}
int main()
{
int k[maxn], a[maxn], rec[maxn];
int fir;
priority_queue<int, vector<int>, less<int> >q;
while(~scanf("%d", &n)){
memset(c, 0, sizeof(c));
memset(k, 0, sizeof(k));
memset(rec, 0, sizeof(rec));
fir = 0;
while(!q.empty()) q.pop();
for (int i = 1; i <= n; i++){
scanf("%d", &a[i]);
if (!fir) fir = a[i];
q.push(a[i]);
add(a[i], 1);
int num = Sum(a[i] - 1);
if (num == i - 1) rec[a[i]] = 1;
else if (num == i - 2) k[q.top()]++;
}
int mx = -1, t = -1;
for (int i = 1; i <= n; i++){
if (k[i] > mx){
mx = k[i];
t = i;
}
}
if (mx == 1){
if (t == fir || rec[t] == 1) t = 1;
}
if (mx == 0){
for (int i = 1; i <= n; i++)
if (i != a[i]){
t = i;
break;
}
}
printf("%d\n", t);
}
return 0;
}