E. Subordinates
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
In the first example it is possible that only the first worker made a mistake. Then:
- the immediate superior of the first worker is the second worker,
- the immediate superior of the third worker is the first worker,
- the second worker is the chief.
目标序列是有一个零,后面是连续的。然后贪心地把后面的数调到前面就好了。
#include <bits/stdc++.h> using namespace std; int n, s, ans; int a[200020]; multiset<int> ss; int main() { #ifdef ULTMASTER freopen("a.in","r",stdin); #endif scanf("%d %d", &n, &s); for (int i = 0; i < n; ++i) scanf("%d", &a[i]); if (a[s - 1] != 0) { a[s - 1] = 0; ans++; } { int zero = 0; for (int i = 0; i < n; ++i) if (a[i] == 0) { if (zero == 0) zero = 1; else a[i] = 1e8; } ss.insert(a, a + n); } int prev = 0; while (!ss.empty()) { auto it = ss.begin(); if ((*it) - prev > 1) { auto itend = (--ss.end()); prev++; ans++; ss.erase(itend); } else { prev = (*it); ss.erase(it); } } printf("%d ", ans); return 0; }