單身狗——排序2

水題檢測鳥發表於2020-10-12

在這裡插入圖片描述
這題用兩個for迴圈會超時;
是個有思想的題;
有一種用三個陣列存情侶互相關係的方法。
具體看程式碼和註釋,超詳細。

#include <bits/stdc++.h>
using namespace std;
#define MAXN 100010
int a[MAXN],b[MAXN],c[MAXN];//用c陣列記錄該人的物件需不需要'關愛'
int in[MAXN],out[MAXN];
int main()
{
    ios::sync_with_stdio(false);//加速
    int n;
    cin>>n;
    for(int i=0; i<n; i++)
    {
        int x,y;
        cin>>x>>y;//輸入一對
        a[x]=y;//a陣列儲存每個人的物件
        a[y]=x;
        b[x]=b[y]=1;//b陣列儲存此人是否有物件
    }
    int m;
    cin>>m;
    for(int i=0; i<m; i++)
    {
        cin>>in[i];//in陣列為輸入的人
        if(b[in[i]])//如果這個人有物件
            c[a[in[i]]]=1;//則這個人的物件不需要關照
    }
    int k=0;
    for(int i=0; i<m; i++)
    {
        if(!c[in[i]])//需要關照的人
            out[k++]=in[i];//將需要關照的人放到out陣列
    }
    sort(out,out+k);//排序
    cout<<k<<endl;//輸出需要關照的人數
    for(int i=0; i<k; i++)
        printf("%05d%c",out[i],i==k-1?'\n':' ');//輸出
    return 0;
}

相關文章