题目描述:
有 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 分钟后池塘的状态。
输入
5
1 2 0 3 4
输出
2 3 1 4 0
说明/提示
样例解释
第一分钟后:0 2 1 3 4
第二分钟后:2 0 1 3 4
第三分钟后:2 3 1 0 4
第四分钟后:2 3 1 4 0
没有优化前:
#include<iostream>
using namespace std;
int a[1000005],w[1000005];
int main()
{
int n,i,j,wi,w0,t;
cin>>n;
for(i=1;i<=n;i++)
{
cin>>a[i];
w[a[i]]=i;
}
for(i=1;i<=n-1;i++)
{
wi=w[i];
w0=w[0];
t=a[wi];
a[wi]=a[w0];
a[w0]=t;
t=w[i];
w[i]=w[0];
w[0]=t;
}
for(i=1;i<=n;i++)
{
cout<<a[i]<<" ";
}
return 0;
}
这时候我们就发现了问题,超过1秒了!
优化后:
#include<iostream>
using namespace std;
int a[1000005];
int main()
{
int n,i;
cin>>n;
for(i=1;i<=n;i++)
{
cin>>a[i];
}
for(i=1;i<=n;i++)
{
if(a[i]==n-1)
{
cout<<0<<" ";
}
else
{
cout<<a[i]+1<<" ";
}
}
return 0;
}
这样就可以了。