洛谷 B3842 [GESP202306 三级] 春游

题目描述

老师带领同学们春游。已知班上有 N 位同学，每位同学有从 0 到 N−1 的唯一编号。到了集合时间，老师确认是否所有同学都到达了集合地点，就让同学们报出自己的编号。到达的同学都会报出自己的编号，不会报出别人的编号，但有的同学很顽皮，会多次报出。你能帮老师找出有哪些同学没有到达吗 ?。

输入格式

输入包含 2 行。第一行包含两个整数 N 和 M，表示班级有 N 位同学，同学们共有 M 次报出编号。约定 2≤N,M≤1000。
第二行包含 M 个整数，分别为 M 次报出的编号。约定所有编号是小于 N 的非负整数。

输出格式

输出一行。如果所有同学都到达，则输出 N；否则由小到大输出所有未到达的同学编号，空格分隔。

输入输出样例

输入 #1

3 3
0 2 1

输出 #1

3

输入 #2

3 5
0 0 0 0 0

输出 #2

1 2

解题思路

这道题比较水，主要在于标记
这道题的核心是找出未到达集合地点的同学编号。思路如下：
1.标记已到达的同学：用一个数组记录哪些同学已经报过到（无论报了多少次）
2.查找未到达的同学：遍历所有可能的编号，找出没有被标记的编号
3.按要求输出结果：如果所有同学都到了，输出班级总人数 N；否则按从小到大输出未到同学的编号

代码解析

#include <bits/stdc++.h>
using namespace std;

const int N=1010;  // 定义常量N为1010，满足题目中N≤1000的要求

int a[N];  // 用于标记同学是否到达，a[i]=1表示编号i的同学已到达
int ans=0;  // 记录未到达同学的数量
int n,m;  // n表示总人数，m表示报数次数

int main(){
    cin>>n>>m;  // 输入总人数和报数次数
    
    // 处理m次报数
    for(int i=0; i<m; i++){
        int p;
        cin>>p;  // 读取报出的编号
        a[p]=1;  // 标记该编号的同学已到达
    }
    
    // 检查所有同学的到达情况
    for(int i=0; i<n; i++){
        if(a[i]==0){  // 如果该编号未被标记（未到达）
            ans++;  // 未到达人数加1
            cout<<i<<" ";  // 输出该编号
        }
    }
    
    // 如果所有同学都到达了
    if(ans==0) {
        cout<<n<<endl;  // 输出总人数n
    }

    return 0;
}

代码特点
使用数组标记法高效记录到达情况，时间复杂度为 O (m+n)
空间复杂度为 O (n)，用于存储到达标记
自然满足了 "从小到大输出未到达同学编号" 的要求，因为我们是从 0 到 n-1 依次检查的
处理了两种边界情况：所有同学都到达和部分同学未到达