B3917 [语言月赛 202401] 小跳蛙

 

题目描述

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

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

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

输入格式

输入共两行。

第一行一个整数 n。
第二行 n 个整数 a1​,a2​,⋯,an​。

输出格式

输出一行 n 个整数 a1​,a2​,⋯,an​。 表示n−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

#include <bits/stdc++.h>
using namespace std;
int a[1000005]={};
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int j=1;j<=n;j++){
        if(a[j]!=n-1) a[j]++;  //第j分钟，第j个数变成0，在下一秒就会变成他加1的数
        else a[j]=0;           //因为只有n-1秒，所以最初为n-1的位置就只能是0
    }
    for(int k=1;k<=n;k++){
        cout<<a[k]<<" ";
    }
    return 0;
}