#include <iostream>
using namespace std;

int main()
{
    int n,m,tmp;
    for (;cin >> n >> m;)
    {
        int arr[10001]{0};
        int flag=0;
        for (int i = 0; i < m;++i)
        {
            cin >> tmp;
            arr[tmp] = 1;
        }
        for (int i = 1;i<=n;++i)
        {
            if(!arr[i])
            {
                cout << i << " ";
                flag = 1;
            } 
        }
        cout<<(flag?"\n":"@\n");
    }
    
}

以範例測資判斷，士兵編號應該是從1開始
我的想法是:
既然測資範圍只到10000
那就開個陣列紀錄回來的士兵號碼就好
把回來的士兵號碼做個記號
答案就輸出沒做記號的士兵號碼就好
在設個變數判斷是不是全部的士兵都有回來
如果有就輸出＠

我還是幫他寫了一下.當然先把測試資料存成檔案,然後讀取處理.雖然看起來正常,但故意少了一個排序步驟,若能補上,才會比較可靠穩定,這個就留給他了.

#!/usr/bin/env python3

counter = 0

with open('ithelp200518_testdata.txt') as fin:
  for line in fin:
    counter = counter + 1
    splst = line.split()
    if counter % 2 == 1:
      n = int(splst[0])
      m = int(splst[1])
    else:
     if n == m:
       print('@')
     else:
       soldierlst = line.split()
       soldierlst = list(map(int, soldierlst))
       soldierset = set(soldierlst)
       stupidset = set([x for x in range(1, n+1)])
       difflst = list(stupidset - soldierset)
       print(*difflst)

執行結果如下圖:

趕快去隔壁「技術文章」區
找一位心原一馬大大
他最近有在刷 LeetCode
可以試著拜託他
順便刷一下林口高中的 ZeroJudge

選我正解

import random
n=6
m=6

member=[random.randint(1,n) for i in range(m) ]
print(member)
#member=[6,1,3,2,5,4]
lists=[]
for i in range(m):
    lists.append(member[i])


numbers=range(1,n+1)
a=set(numbers).difference(lists)
if(a==set()):
    print('@')
else:   
    print(a)