problemas-para-competicao-pibiti-arthur-dolival/flowers/src/generator.cpp

#include "testlib.h"
#include <bits/stdc++.h>

using namespace std;

const int MIN_N = 1;
const int MAX_N = 1e4;
const int MIN_M = 1;
const int MAX_M = 1e3;

const int rnd_test_n = 100;

template <typename T> void append(vector<T> &dest, const vector<T> &orig) {
	dest.insert(dest.end(), orig.begin(), orig.end());
}

string output_tc(int x, int y) {
	ostringstream oss;
    oss << x << " " << y << endl;
	return oss.str();
}

vector<string> generate_sample_tests() {
	vector<string> tests;
    tests.push_back(output_tc(1, 1));
    tests.push_back(output_tc(2, 2));
    tests.push_back(output_tc(2, 1));
	return tests;
}

vector<string> generate_manual_tests() {
	vector<string> tests;
	tests.push_back(output_tc(100, 20));
	tests.push_back(output_tc(4, 100));
	return tests;
}

string rnd_test(int i){
    int min_n = MIN_N;
    int max_n = MAX_N;
    
    if(i<rnd_test_n / 3){
        max_n = MAX_N/5;
    }
    else if(i<rnd_test_n / 2){
        max_n = MAX_N/2;
    }

    int x = rnd.next(min_n, max_n);
    int y = rnd.next(MIN_M, MAX_M);
    return(output_tc(x, y));
}

vector<string> generate_random_tests() { 
    vector<string> tests;
    for (int i = 0; i < rnd_test_n; i++){
        tests.push_back(rnd_test(i));
    } 
    return tests;
}

string extreme_test_1(){
    return(output_tc(MAX_N, MIN_M));
}
string extreme_test_2(){
    return(output_tc(MAX_N, MAX_M));
}
string extreme_test_3(){
    return(output_tc(MIN_N, MAX_M));
}

vector<string> generate_extreme_tests(){
    vector<string> tests;   
    tests.push_back(extreme_test_1());
    tests.push_back(extreme_test_2());
    tests.push_back(extreme_test_3());
    return tests;
}

int main(int argc, char *argv[]) {
	registerGen(argc, argv, 1);
	vector<string> tests;
	size_t test = 0;
	append(tests, generate_sample_tests());
	append(tests, generate_manual_tests());
	append(tests, generate_random_tests());
	append(tests, generate_extreme_tests());
	for (const auto &t : tests) {
		startTest(++test);
		cout << t;
	}
	return 0;
}