题目描述
格格有一个长度为 nnn 的排列 ppp,但她不记得 ppp 具体的样子,她只记得数组 aaa。
其中:ai=gcd(p1,p2,...,pi)a_i = gcd(p_1, p_2,...,p_i)ai=gcd(p1,p2,...,pi),也就是说,aia_iai 表示排列 ppp 中前 iii 个数字的最大公约数。
现在,她希望小苯将排列 ppp 复原出来,请你帮帮他吧。
(但有可能无解,这意味着格格给出的 aaa 数组可能是不正确的,此时输出 −1-1−1 即可。)
代码
#include<bits/stdc++.h>
using namespace std;
const int N=200010;
int a[N];
int st[N],p[N];
int n;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)cin>>a[i];
int flag=0;
for(int i=1;i<n;i++)
{
if(a[i]%a[i+1])flag=1;
}
if(flag)cout<<"-1"<<endl;
else{
flag=0;
for(int i=1;i<=n;i++)
{
if(a[i]!=a[i-1])
{
p[i]=a[i];
st[a[i]]=1;
}else{
for(int j=p[i-1]+a[i];j<=n;j+=a[i])
{
if(st[j]==0&&__gcd(a[i-1],j)==a[i])
{
st[j]=1;
p[i]=j;
break;
}
}
if(p[i]==0)flag=1;
}
}
if(flag)cout<<"-1"<<endl;
else{
for(int i=1;i<=n;i++)
{
if(i!=1)cout<<" ";
cout<<p[i];
}
cout<<endl;
}
}
}
思路
该题我写的时候主要是想错了,想的是通过前一个数来推出后一个数,应该填的是该数的倍数,那又有一个问题,如果直接枚举该数的倍数,遇到全是1的情况,那就会超时,所以如果a[i]!=a[i-1],直接就可以填a[i],如果a[i]==a[i-1],那么我们从p[i-1]+a[i]开始枚举,找到a[i]的倍数的最小值。