There are n workers in a company, each of them has a unique id from 1 to n. Exaclty one of them is a chief, his id is s. Each worker except the chief has exactly one immediate superior.
There was a request to each of the workers to tell how how many superiors (not only immediate). Worker's superiors are his immediate superior, the immediate superior of the his immediate superior, and so on. For example, if there are three workers in the company, from which the first is the chief, the second worker's immediate superior is the first, the third worker's immediate superior is the second, then the third worker has two superiors, one of them is immediate and one not immediate. The chief is a superior to all the workers except himself.
Some of the workers were in a hurry and made a mistake. You are to find the minimum number of workers that could make a mistake.
The first line contains two positive integers n and s (1 ≤ n ≤ 2·105, 1 ≤ s ≤ n) — the number of workers and the id of the chief.
The second line contains n integers a1, a2, ..., an (0 ≤ ai ≤ n - 1), where ai is the number of superiors (not only immediate) the worker with id i reported about.
Print the minimum number of workers that could make a mistake.
3 2 2 0 2
1
5 3 1 0 0 4 1
2
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int maxn = 2e5 + 7;
int a[maxn];
int main()
{
int n, m, x, flag = 0, record = 0;
cin >> n >> m;
for(int i = 1; i <= n; i ++){
cin >> x;
if(i == m && x != 0) flag = 1; /// 判断主任的上司是否为0,如果不为0那么这种情况肯定错误的,不用把其记录到a[x]中
else a[x] ++;
}
int ans = 0, sum = 0, num = n; /// ans 是错误人员数,sum已经访问过的人员的个数
for(int i = 0; i < n; i ++){
if(i == 0){ /// 特判上司数为0的情况
sum ++;
if(flag){ /// 此处是主任的上司不为为0
ans += (a[i] + 1);
record += a[i];
}
else{
ans += (a[i] - 1);
record += (a[i] - 1);
} /// record是记录的多余的上司数为0的员工的个数
}
else{
if(a[i] == 0){ /// 当上司数为i的员工不存在时
sum ++;
if(record > 0) record --; /// 此处是用上司数为0的多余的员工来补足上司数为i的员工
else{ /// 此处是用上司数最大的员工来补足上司数为i的员工
ans ++;
while(a[num] == 0) num --;
a[num] --;
}
}
else sum += a[i];
}
if(sum == n) break;
}
cout << ans << endl;
return 0;
}