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 = ;
    while(!isdigit(c)) c = getchar();
    while(isdigit(c))
    {
        x = x *  + c - '';
        c = getchar();
    }
}

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

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

int flag[];

queue<int>Q;

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

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

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

    int wei=n;

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

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

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

    cout<<ans<<endl;

    return ;
}