CodeForces 738E Subordinates

排序，构造。

相当于告诉我们一棵树$n$个节点，每个节点在哪一层，至少需要移动多少个节点，才能让这些节点变成一棵树。

按照层次排个序移动一下就可以了，优先选择那些不是$s$但是层次是$0$的节点，如果没有，那么再选择层次最高的。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<ctime>
#include<iostream>
using namespace std;
typedef long long LL;
const double pi=acos(-1.0);
void File()
{
freopen("D:\\in.txt","r",stdin);
freopen("D:\\out.txt","w",stdout);
}
template <class T>
inline void read(T &x)
{
char c = getchar();
x = 0;
while(!isdigit(c)) c = getchar();
while(isdigit(c))
{
x = x * 10 + c - '0';
c = getchar();
}
}

struct X
{
int id,c;
}t[200010];
int n,s,ans;

bool cmp(X a, X b)
{
return a.c<b.c;
}

int flag[200010];

queue<int>Q;

int main()
{
cin>>n>>s;
for(int i=1;i<=n;i++)
{
t[i].id=i;
cin>>t[i].c;
if(t[i].c==0&&i!=s)
{
flag[i]=1;
ans++;
Q.push(i);
}
}

if(t[s].c!=0) ans++,t[s].c=0;

sort(t+1,t+1+n,cmp);

int wei=n;

for(int i=1;i<=n;i++)
{
if(flag[t[i].id]==1) continue;
if(t[i].c==t[i-1].c) continue;
if(t[i].c==t[i-1].c+1) continue;

int need=t[i].c-t[i-1].c-1;
int f=0;

while(1)
{
f++;
if(!Q.empty()) Q.pop();
else
{
if(wei<i) break;
flag[t[wei].id]=1;
wei--; ans++;
if(wei<i) break;
}
if(f==need) break;
}
if(wei<i) break;
}

cout<<ans<<endl;

return 0;
}