problem(A1054): fix generator

Author: StableAgOH

problems/A/A1054/A1054.md

@@ -22,15 +22,13 @@ Quento 游戏的规则非常简单，给定一个“Quento 棋盘格”，你会
 
 # 输入格式
 
-输入由多个测试用例组成。第一行包含一个整数 $t(1 \leq t \leq {10}^5)$，代表测试用例的数量。测试用例的描述如下：
+输入由多个测试用例组成。第一行包含一个整数 $t$，代表测试用例的数量。测试用例的描述如下：
 
-前三行，每行三个整数，代表给定的“Quento 棋盘格”。黑色数字格上的整数代表该格子上的数字，保证这个数字的绝对值小于等于 $20$。白色运算符格上的整数只会为 $1$ 或 $-1$：若为 $1$，则代表该格子上的运算符为 `+`；若为 $-1$，则代表该格子上的运算符为 `-`。
+前三行，每行三个整数，代表给定的“Quento 棋盘格”。黑色数字格上的整数代表该格子上的数字。白色运算符格上若为 $1$，则代表该格子上的运算符为 `+`，若为 $-1$，则代表该格子上的运算符为 `-`。
 
-第四行，一个整数 $m(1 \leq m \leq {10}^5)$，代表询问的数量。
+第四行，一个整数 $m$，代表询问的数量。
 
-接下来 $m$ 行，每行两个整数 $x_i, y_i(|x_i| \leq 100$，$1 \leq y_i \leq 5)$，代表一次询问。
-
-保证 $\sum m \leq 2 \times {10}^5$。
+接下来 $m$ 行，每行两个整数 $x_i, y_i$，代表一次询问。
 
 # 输出格式
 
@@ -84,3 +82,9 @@ Yes
 第三个询问可以通过以下方式达成（$5+6-7-8+3=-1$）：
 
 ![](file://quento5.jpg)
+
+【数据规模与约定】
+
+对于 $30 \%$ 的数据，$t=1$，$m \leq 2000$。
+
+对于 $100 \%$ 的数据，$1 \leq t \leq {10}^5$，$1 \leq m \leq {10}^5$，$\sum m \leq 5 \times {10}^5$，$|x_i| \leq 100$，$1 \leq y_i \leq 5$。保证黑色数字格上只会为绝对值小于等于 $20$ 的整数，白色运算符格上只会为 $1$ 或 $-1$。

problems/A/A1054/gen.cpp

@@ -16,47 +16,65 @@ int main(int argc, char const* argv[])
     {
         ofstream fout(to_string(_t)+".in");
         // ==============================
-        switch(_t)
+        auto f = [&]()
         {
-        case 1: case 2:
-            fout<<200<<'\n';
-            for(int o=0;o<200;o++)
-            {
-                fout<<rndx()<<' '<<1<<' '<<rndx()<<'\n';
-                fout<<1<<' '<<rndx()<<' '<<1<<'\n';
-                fout<<rndx()<<' '<<1<<' '<<rndx()<<'\n';
-                fout<<1000<<'\n';
-                for(int j=1;j<=5;j++)
-                    for(int i=-100;i<=100;i++)
-                        fout<<i<<' '<<j<<'\n';
-            }
-            break;
-        case 3: case 4:
-            fout<<200<<'\n';
-            for(int o=0;o<200;o++)
-            {
-                fout<<rndx()<<' '<<-1<<' '<<rndx()<<'\n';
-                fout<<-1<<' '<<rndx()<<' '<<-1<<'\n';
-                fout<<rndx()<<' '<<-1<<' '<<rndx()<<'\n';
-                fout<<1000<<'\n';
-                for(int j=1;j<=5;j++)
-                    for(int i=-100;i<=100;i++)
-                        fout<<i<<' '<<j<<'\n';
-            }
-            break;
-        default:
-            fout<<200<<'\n';
-            for(int o=0;o<200;o++)
+            for(int j=1;j<=5;j++)
+                for(int i=-100;i<=100;i++)
+                    fout<<i<<' '<<j<<'\n';
+        };
+        auto one = [&]() { fout<<1005<<'\n'; f(); };
+        auto two = [&]() { fout<<2010<<'\n'; f(); f(); };
+        auto b_pos = [&]()
+        {
+            fout<<rndx()<<' '<<1<<' '<<rndx()<<'\n';
+            fout<<1<<' '<<rndx()<<' '<<1<<'\n';
+            fout<<rndx()<<' '<<1<<' '<<rndx()<<'\n';
+        };
+        auto b_neg = [&]()
+        {
+            fout<<rndx()<<' '<<-1<<' '<<rndx()<<'\n';
+            fout<<-1<<' '<<rndx()<<' '<<-1<<'\n';
+            fout<<rndx()<<' '<<-1<<' '<<rndx()<<'\n';
+        };
+        auto b_rnd = [&]()
+        {
+            fout<<rndx()<<' '<<rndo()<<' '<<rndx()<<'\n';
+            fout<<rndo()<<' '<<rndx()<<' '<<rndo()<<'\n';
+            fout<<rndx()<<' '<<rndo()<<' '<<rndx()<<'\n';
+        };
+        int t;
+        if(_t<=0.3*CASES) t = 1;
+        else t = 200;
+        fout<<t<<'\n';
+        while(t--)
+        {
+            switch(_t)
             {
-                fout<<rndx()<<' '<<rndo()<<' '<<rndx()<<'\n';
-                fout<<rndo()<<' '<<rndx()<<' '<<rndo()<<'\n';
-                fout<<rndx()<<' '<<rndo()<<' '<<rndx()<<'\n';
-                fout<<1000<<'\n';
-                for(int j=1;j<=5;j++)
-                    for(int i=-100;i<=100;i++)
-                        fout<<i<<' '<<j<<'\n';
+            case 1:
+                b_pos();
+                one();
+                break;
+            case 2:
+                b_neg();
+                one();
+                break;
+            case 3:
+                b_rnd();
+                one();
+                break;
+            case 4: case 5:
+                b_pos();
+                two();
+                break;
+            case 6: case 7:
+                b_neg();
+                two();
+                break;
+            default:
+                b_rnd();
+                two();
+                break;
             }
-            break;
         }
         // ==============================
         fout.flush();