Cod sursa(job #2918450)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 11 august 2022 15:42:24
Problema Progresie Scor 0
Compilator cpp-64 Status done
Runda Arhiva ICPC Marime 0.7 kb
#include<bits/stdc++.h>
using namespace std;
ifstream F("progresie.in");
ofstream G("progresie.out");
int t;
long long n,r,l,j,k,i,y;
int main()
{
    for(F>>t;t;--t) {
        for(F>>n>>r,i=1;;++i)
            for(j=i*(i-1)+1;j<=i*i;) {
                for(k=0;k<n;++k) {
                    y=j+k*r,l=sqrt(y);
                    if(y==l*l)
                        continue;
                    ++l;
                    if(l*(l-1)+1<=y&&y<=l*l)
                        continue;
                    j+=(l-1)*l+1-y;
                    break;
                }
                if(k==n)
                    goto E;
            }
        E:
            G<<j<<'\n';
    }
    return 0;
}