B3917 [语言月赛 202401] 小跳蛙

题目描述

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

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

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

输入格式

输入共两行。

第一行一个整数 𝑛n。
第二行 𝑛n 个整数 𝑎1,𝑎2,⋯ ,𝑎𝑛a1​,a2​,⋯,an​。

输出格式

输出一行 𝑛n 个整数 𝑎1,𝑎2,⋯ ,𝑎𝑛a1​,a2​,⋯,an​。 表示 𝑛−1n−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%50% 的数据，满足 1≤𝑛≤1031≤n≤103。

对于 100%100% 的数据，满足 1≤𝑛≤1061≤n≤106，保证序列 𝑎a 是一个 0∼𝑛−10∼n−1 这些数字的排列。

解题步骤：

首先，有题知，一共有n-1个小跳蛙，数组量为1000000，所以数组定义在主函数外。

其次，我们要用for循环来寻找编号为i的小跳蛙和空位，并与其互换。

最后输出。

代码实现。

#include<iostream>
int a[1000005];
using namespace std;
int main(){
    int n,i,j,wi,w0,t;
    cin>>n;
    for(i=1;i<=n;i++){
        cin>>a[i];
    }
    for(i=1;i<=n-1;i++){
        for(j=1;j<=n;j++){
            if(a[j]==i){
                wi=j;
            }
            if(a[j]==0){
                w0=j;
            }
        }
        t=wi;
        wi=w0;
        w0=t;
    }
    for(i=1;i<=n;i++){
        cout<<a[i]<<" ";
    }
    return 0;
}

but

你会发现，洛谷上却只有50分，那是因为，在寻找i和空位时，次数过多，导致时间不够。

所以，要在那另一个数组存a[i]所在的i的位置，在对其进行更新，就可以了