CF 1512 G. Short Task

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e7+5;
int p[N],d[N],ans[N];
void solve(){
    
    
    int n;
    scanf("%d",&n);
    printf("%d\n",ans[n]);
}
int main()
{
    
    
    fill(p,p+N,-1);
    p[1]=1;
    for(int i=2;i*i<N;i++)
    {
    
    
        if(p[i]==-1)
        {
    
    
            p[i]=i;
            for(int j=i;j<N;j+=i)p[j]=i;
        }
    }
    d[1]=1;
    for(int i=2;i<N;i++)
    {
    
    
        if(p[i]==-1)d[i]=i+1;
        else 
        {
    
    
            int t=i;
            d[i]=1;
            while(t%p[i]==0)
            {
    
    
                d[i]=d[i]*p[i]+1;
                t/=p[i];
            }
            d[i]*=d[t];
        }
    }
    fill(ans,ans+N,-1);
    for(int i=N-1;i>=1;i--)if(d[i]<N)ans[d[i]]=i;
    int T;
    scanf("%d",&T);
    while(T--)solve();
    return 0;
}