Files
problemas-para-competicao-p…/delete-and-earn/src/generator.cpp
2025-11-06 12:15:56 -03:00

99 lines
2.1 KiB
C++

#include "testlib.h"
#include <bits/stdc++.h>
using namespace std;
const int MIN_N = 1;
const int MAX_N = 20000;
const int MIN_NI = 1;
const int MAX_NI = 10000;
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(const vector<int>& nums) {
ostringstream oss;
oss << nums.size() << endl;
for (int i = 0; i < nums.size(); i++) {
if (i != 0) oss << " ";
oss << nums[i];
}
oss << endl;
return oss.str();
}
vector<string> generate_sample_tests() {
vector<string> tests;
tests.push_back(output_tc({3, 4, 5}));
tests.push_back(output_tc({2, 2, 3, 3, 3, 4}));
tests.push_back(output_tc({1, 2, 3, 4, 5}));
return tests;
}
vector<string> generate_manual_tests() {
vector<string> tests;
tests.push_back(output_tc({1, 1, 1, 1, 1, 1, 1, 1, 1, 1}));
tests.push_back(output_tc({2, 5, 9, 11, 17, 28, 100, 123, 1230}));
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 = 5;
}
else if(i<rnd_test_n / 2){
max_n = 20;
}
int x = rnd.next(min_n, max_n);
vector<int> nums(x);
for (int i = 0; i < x; i++) {
nums[i] = rnd.next(MIN_NI, MAX_NI);
}
return(output_tc(nums));
}
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(){
vector<int> nums(MAX_N);
for (int i = 0; i < MAX_N; i++) {
nums[i] = MAX_NI;
}
return(output_tc(nums));
}
vector<string> generate_extreme_tests(){
vector<string> tests;
tests.push_back(extreme_test_1());
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;
}