Modify generator

Author: ShinyaKato

battle_royal/tests/generator.cpp

@@ -6,41 +6,81 @@
 #include "constraints.h"
 using namespace std;
 
-double normal(double mu, double sigma) {
-  const double PI = acos(-1);
-  double z = sqrt(-2.0*log(rnd.next(1.0))) * sin(2.0 * PI * rnd.next(1.0));
-  return mu + z * sigma;
+vector<int> random_case(int kill_sum, int killers) {
+  int remain = kill_sum - killers;
+  vector<int> K(PLAYERS);
+  for(int i = 0; i < killers; i++) K[i] = 1;
+  for(int i = 0; i < remain; i++) {
+    K[rnd.next(killers)]++;
+  }
+  shuffle(K.begin(), K.end());
+  return K;
 }
 
-void generate(const string &name, int T) {
-  ofstream ofs(name);
+vector<int> no_kill_case() {
+  vector<int> K(PLAYERS);
+  return K;
+}
 
-  ofs << T << endl;
+void write_data_set(const string &name, vector< vector<int> > &data_set) {
+  int T = data_set.size();
+  ofstream ofs(name);
+  ofs << T << "\n";
   for(int t = 0; t < T; t++) {
-    int K[PLAYERS];
-    int remain = PLAYERS - 1;
-    double mu = rnd.next(1.0, 5.0);
-    double sigma = rnd.next(0.0, 5.0);
     for(int i = 0; i < PLAYERS; i++) {
-      K[i] = min(max((int) normal(mu, sigma), 0), remain);
-      remain -= K[i];
+      ofs << data_set[t][i] << (i + 1 != PLAYERS ? " " : "\n");
     }
+  }
+}
 
-    random_device seed_gen;
-    mt19937 engine(seed_gen());
-    shuffle(K, K + PLAYERS, engine);
+void generate_small() {
+  vector< vector<int> > data_set;
+  data_set.push_back(random_case(32, 15));
+  data_set.push_back(random_case(54, 3));
+  data_set.push_back(random_case(82, 10));
+  data_set.push_back(random_case(98, 87));
+  data_set.push_back(random_case(77, 2));
+  data_set.push_back(no_kill_case());
+  write_data_set("small.in", data_set);
+}
 
-    for(int i = 0; i < PLAYERS; i++) {
-      ofs << K[i] << (i + 1 != PLAYERS ? " " : "\n");
-    }
+void generate_large() {
+  vector< vector<int> > data_set;
+  for(int t = 0; t < 90; t++) {
+    int kill_sum = rnd.next(PLAYERS - 2) + 1;
+    int killers = rnd.next(kill_sum) + 1;
+    data_set.push_back(random_case(kill_sum, killers));
+  }
+  data_set.push_back(random_case(32, 15));
+  data_set.push_back(random_case(54, 3));
+  data_set.push_back(random_case(82, 10));
+  data_set.push_back(random_case(98, 87));
+  data_set.push_back(random_case(77, 2));
+  data_set.push_back(random_case(99, 99));
+  data_set.push_back(random_case(99, 1));
+  data_set.push_back(random_case(5, 4));
+  data_set.push_back(random_case(42, 7));
+  data_set.push_back(no_kill_case());
+  shuffle(data_set.begin(), data_set.end());
+  write_data_set("large.in", data_set);
+}
+
+void generate(const string &name, int T) {
+  vector< vector<int> > data_set;
+  for(int t = 0; t < T; t++) {
+    int kill_sum = rnd.next(PLAYERS - 2) + 1;
+    int killers = rnd.next(kill_sum) + 1;
+    data_set.push_back(random_case(kill_sum, killers));
   }
+  shuffle(data_set.begin(), data_set.end());
+  write_data_set(name, data_set);
 }
 
 int main(int argc, char* argv[]) {
   registerGen(argc, argv, 1);
 
-  generate("small.in", 10);
-  generate("large.in", MAX_T);
+  generate_small();
+  generate_large();
 
   for(int i = 0; i < 10; i++) {
     generate("random_test" + to_string(i) + ".in", MAX_T);