|
| 1 | +#include <fstream> |
| 2 | +#include <string> |
| 3 | +#include <algorithm> |
| 4 | +#include <random> |
| 5 | +#include "testlib.h" |
| 6 | +#include "constraints.h" |
| 7 | +using namespace std; |
| 8 | + |
| 9 | +vector<int> random_case(int kill_sum, int killers) { |
| 10 | + int remain = kill_sum - killers; |
| 11 | + vector<int> K(PLAYERS); |
| 12 | + for(int i = 0; i < killers; i++) K[i] = 1; |
| 13 | + for(int i = 0; i < remain; i++) { |
| 14 | + K[rnd.next(killers)]++; |
| 15 | + } |
| 16 | + shuffle(K.begin(), K.end()); |
| 17 | + return K; |
| 18 | +} |
| 19 | + |
| 20 | +vector<int> no_kill_case() { |
| 21 | + vector<int> K(PLAYERS); |
| 22 | + return K; |
| 23 | +} |
| 24 | + |
| 25 | +void write_data_set(const string &name, vector< vector<int> > &data_set) { |
| 26 | + int T = data_set.size(); |
| 27 | + ofstream ofs(name); |
| 28 | + ofs << T << "\n"; |
| 29 | + for(int t = 0; t < T; t++) { |
| 30 | + for(int i = 0; i < PLAYERS; i++) { |
| 31 | + ofs << data_set[t][i] << (i + 1 != PLAYERS ? " " : "\n"); |
| 32 | + } |
| 33 | + } |
| 34 | +} |
| 35 | + |
| 36 | +void generate_small() { |
| 37 | + vector< vector<int> > data_set; |
| 38 | + data_set.push_back(random_case(32, 15)); |
| 39 | + data_set.push_back(random_case(54, 3)); |
| 40 | + data_set.push_back(random_case(82, 10)); |
| 41 | + data_set.push_back(random_case(98, 87)); |
| 42 | + data_set.push_back(random_case(77, 2)); |
| 43 | + data_set.push_back(no_kill_case()); |
| 44 | + write_data_set("small.in", data_set); |
| 45 | +} |
| 46 | + |
| 47 | +void generate_large() { |
| 48 | + vector< vector<int> > data_set; |
| 49 | + for(int t = 0; t < 90; t++) { |
| 50 | + int kill_sum = rnd.next(PLAYERS - 2) + 1; |
| 51 | + int killers = rnd.next(kill_sum) + 1; |
| 52 | + data_set.push_back(random_case(kill_sum, killers)); |
| 53 | + } |
| 54 | + data_set.push_back(random_case(32, 15)); |
| 55 | + data_set.push_back(random_case(54, 3)); |
| 56 | + data_set.push_back(random_case(82, 10)); |
| 57 | + data_set.push_back(random_case(98, 87)); |
| 58 | + data_set.push_back(random_case(77, 2)); |
| 59 | + data_set.push_back(random_case(99, 99)); |
| 60 | + data_set.push_back(random_case(99, 1)); |
| 61 | + data_set.push_back(random_case(5, 4)); |
| 62 | + data_set.push_back(random_case(42, 7)); |
| 63 | + data_set.push_back(no_kill_case()); |
| 64 | + shuffle(data_set.begin(), data_set.end()); |
| 65 | + write_data_set("large.in", data_set); |
| 66 | +} |
| 67 | + |
| 68 | +void generate(const string &name, int T) { |
| 69 | + vector< vector<int> > data_set; |
| 70 | + for(int t = 0; t < T; t++) { |
| 71 | + int kill_sum = rnd.next(PLAYERS - 2) + 1; |
| 72 | + int killers = rnd.next(kill_sum) + 1; |
| 73 | + data_set.push_back(random_case(kill_sum, killers)); |
| 74 | + } |
| 75 | + shuffle(data_set.begin(), data_set.end()); |
| 76 | + write_data_set(name, data_set); |
| 77 | +} |
| 78 | + |
| 79 | +int main(int argc, char* argv[]) { |
| 80 | + registerGen(argc, argv, 1); |
| 81 | + |
| 82 | + generate_small(); |
| 83 | + generate_large(); |
| 84 | + |
| 85 | + for(int i = 0; i < 10; i++) { |
| 86 | + generate("random_test" + to_string(i) + ".in", MAX_T); |
| 87 | + } |
| 88 | + |
| 89 | + return 0; |
| 90 | +} |
0 commit comments