feat: new problem formatted

rod-cutting/src/generator.cpp

@@ -0,0 +1,125 @@
+#include "testlib.h"
+#include <bits/stdc++.h>
+
+using namespace std;
+
+const int MIN_N = 1;
+const int MAX_N = 1e3;
+const int MIN_NI = 1;
+const int MAX_NI = 1e4;
+
+const int rnd_test_n = 40;
+
+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({1, 5, 8, 9, 10, 17, 17, 20}));
+    tests.push_back(output_tc({1, 2, 3, 5}));
+    tests.push_back(output_tc({3, 5, 8, 9, 10, 17, 17, 20}));
+    return tests;
+}
+
+vector<string> generate_manual_tests()
+{
+    vector<string> tests;
+    tests.push_back(output_tc({MIN_NI, MIN_NI, MIN_NI, MIN_NI}));
+    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 / 3;
+    }
+    else if (i < rnd_test_n / 2)
+    {
+        max_n = MAX_N / 2;
+    }
+
+    int n = rnd.next(min_n, max_n);
+    vector<int> nums(n);
+    for (int i = 0; i < n; 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] = MIN_NI;
+    }
+    return (output_tc(nums));
+}
+
+string extreme_test_2()
+{
+    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());
+    tests.push_back(extreme_test_2());
+    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;
+}