Cod sursa(job #3310099)

Utilizator popescu_georgePopescu George popescu_george Data 11 septembrie 2025 18:03:09
Problema Progresie Scor 100
Compilator cpp-64 Status done
Runda Arhiva ICPC Marime 0.78 kb
#include<fstream>
#include<cmath>
using namespace std;
ifstream cin("progresie.in");
ofstream cout("progresie.out");
int main()
{
    short t;
    for(cin>>t;t--;) {
        short n;
        unsigned r;
        cin>>n>>r;
        for(int i=1;;++i) {
            short j=1;
            long long v=1LL*i*i-i+1;
            for(int e=i-1;j<n;++j) {
                long long l=v+j*r;
                int b=sqrt(l-1)+1;
                long long y=1LL*b*b,x=y-b+1;
                if(l<x)
                    v+=x-l,e+=l-x;
                else if(y-l<e)
                    e=y-l;
                if(l+e<x)
                    break;
            }
            if(j==n) {
                cout<<v<<'\n';
                break;
            }
        }
    }
    return 0;
}