hi.小跳蛙

[语言月赛 202401] 小跳蛙

题目描述

有 $n-1$ 只小跳蛙在池塘中，依次被编号为 $1,2,\cdots ,n-1$。池塘里有 $n$ 个位置，每一个位置上有一个数字 $a_i$。如果 $a_i=0$，则表示这个位置是一个空位；否则表示这个位置上存在一个编号为 $a_i$ 的小跳蛙。

接下来的 $n-1$ 分钟，小跳蛙们将进行跳跃。第 $i$ 分钟，编号为 $i$ 的小跳蛙将跳到空位上。

请你输出 $n-1$ 分钟后池塘中每个位置的数字，即每个位置是否为空、小跳蛙编号是多少。

输入格式

输入共两行。

第一行一个整数 $n$。
第二行 $n$ 个整数 $a_1,a_2,\cdots ,a_n$。

输出格式

输出一行 $n$ 个整数 $a_1,a_2,\cdots ,a_n$。 表示 $n-1$ 分钟后池塘的状态。

样例 #1

样例输入 #1

5
1 2 0 3 4

样例输出 #1

2 3 1 4 0

提示说明

样例解释 #1

• 第一分钟后：0 2 1 3 4
• 第二分钟后：2 0 1 3 4
• 第三分钟后：2 3 1 0 4
• 第四分钟后：2 3 1 4 0

因此最终池塘的状态为 2 3 1 4 0

数据规模与约定

对于 $50\%$ 的数据，满足 $1\le n\le 10^3$。

对于 $100\%$ 的数据，满足 $1\le n\le 10^6$，保证序列 $a$ 是一个 $0\sim n-1$ 这些数字的排列。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll N=1e6+10;
ll a[N];

int main()
{
    ll n ;
    cin>>n;

    for(ll i=1;i<=n;i++)
        cin>>a[i];
    
    for(ll i=1;i<=n;i++)
    {
        if(a[i]!=n-1)
            cout<<a[i]+1<<" ";
        else
            cout<<0<<" ";
    }
    
    return 0;
}