Cod sursa(job #1761071)

Utilizator CataFetoiuFetoiu Catalin CataFetoiu Data 21 septembrie 2016 19:11:44
Problema Progresie Scor 0
Compilator cpp Status done
Runda Arhiva ICPC Marime 1.87 kb
#include<cstdio>
#include<iostream>
#include<string>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<sstream>
#include<limits.h>
#include<math.h>
#include<list>
#include<queue>
#include<map>

using namespace std;

typedef pair<int, int> ii;
typedef long long int64;

#define Max 3100
#define INF 2000000000

int debug;

/* last element in union */
int64 f(int64 seq)
{
	return seq * (seq+1);
}

/* first element in sequence */
int64 first(int64 seq)
{
	return f(seq-1) + 1;
}

/* last element in sequence */
int64 last(int64 seq)
{
	return first(seq) + seq - 1;
}

/* finds min j such that first(j) >= a */
int64 find(int64 a)
{
	return 1 + floor(sqrt(a));
}

int main()	
{
	
	FILE* fin = freopen("progresie.in", "r", stdin);
	FILE* fout = freopen("progresie.out", "w", stdout);

	int T; cin >> T;
	while(T--)
	{

	int N, R;
	cin >> N >> R;
	if(N == 1)
		cout << "1" << endl;
	else
	{
		int64 res = first(1 + (N-1) * R);
		bool found = false;
		
		int block = 0;
		while(true)
		{
			block++;
			if(block >= R)
				break;

			int64 curentStart = first(block);
			int64 curentEnd = last(block);

			//cout << block << endl;

			int64 j;
			bool foundChain = true;
			for(int i = 0; i < N; i++)
			{
				int64 a = curentStart + R; /* max reachable interval from sequence i */
 				int64 b = curentEnd + R;

				//cout << curentStart << " " << curentEnd << endl;

				j = find(a);
				if(first(j) > b)
				{
					foundChain = false;
					break;
				}
				else /* first(j) is <= b */
				{
					if(first(j) < a)
						curentStart = a;
					else
						curentStart = first(j);
					if(last(j) <= b)
						curentEnd = last(j);
					else
						curentEnd = b;
				}
			}

			if(foundChain)
			{
				cout << min(res, curentStart - N * R) << endl;
				found = true;
				break;
			}
		}
		if(!found)
			cout << res << endl;
	
	}

	}

	fclose(fin);
	fclose(fout);

	return 0;
}