Kernel Knights (Gym - 101480K)

2023-03-09 16:52:50 浏览数 (2)

题目链接

代码语言:javascript复制
#include <bits/stdc  .h>

using namespace std;
typedef long long ll;
int a[200005];   //存放原始数据
int vis[200005]; //标记选的对手
int b[200005]; //答案序列
queue<int>q; //把所有能够选的都存一次,接着遍历与它有关系的。
int main()
{
    int n,x;
    scanf("%d",&n);
    memset(vis,0,sizeof(vis));  //初始化
    memset(b,0,sizeof(b));      
    //memset(a,0,sizeof(a));
    for(int i = 1; i <= 2*n; i   )
    {
        scanf("%d",&a[i]);
        vis[a[i]]  ;
    }
    for(int i = 1; i <= 2*n; i   )  // 把能够选的放入到队列中,遍历与它相连的是否可以选上
    {
        if(vis[i] == 0)
            q.push(i);
    }
    while(!q.empty())
    {
        int x = q.front();
        q.pop();
        b[x] = 1;    // 表示可以选择放入到S中
        if(b[a[x]]==-1)   // 如果x被选上了,那么x所选的对手被处理过一次不能选,跳过就可以
            continue;    //因为不跳出,vis会再重复减一次,导致答案会增多或者减少一些
        b[a[x]] = -1;  
        vis[a[a[x]]] --;   // x选的对手y,y选的对手要减1,因为x被选进S中,y现在就相当于x的位置,y的选的对手就可以放进S中
        if(vis[a[a[x]]] == 0)  //当vis[z = a[a[x]]]为0,说明没有人挑战z,所以z要放进S中
            q.push(a[a[x]]);
    }
    for(int i=1; i<=2*n; i  )
    {
        if(i<=n&&b[i]>=0)  // 在第一房间(排)中,只要符合不与下面的在一个集合中就可以选择
            printf("%d ",i);
        else if(b[i]==1)
            printf("%d ",i);
    }
    printf("n");
    return 0;
}

0 人点赞