Merge pull request yosupo06#1240 from NachiaVivias/issue-1239

Sqrt Polynomial の checker で値の範囲をチェック
maspypy · Sep 10, 2024 · e07ff41 · e07ff41
2 parents 0b83c96 + cdbc469
commit e07ff41
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/polynomial/sqrt_of_formal_power_series/checker.cpp b/polynomial/sqrt_of_formal_power_series/checker.cpp
@@ -2,6 +2,7 @@
 #include <iostream>
 #include <cassert>
 #include "testlib.h"
+#include "params.h"
 
 using namespace std;
 using uint = unsigned int;
@@ -113,12 +114,12 @@ bool has_sqrt(vector<Mint> pol) {
 
 bool read_ans(InStream& stream, vector<Mint> expect) {
     int n = int(expect.size());
-    int first = stream.readInt();
+    int first = stream.readInt(-1, MOD - 1);
     if (first == -1) return false;    
     vector<Mint> pol(n);
     pol[0] = first;
     for (int i = 1; i < n; i++) {
-        pol[i] = stream.readInt();
+        pol[i] = stream.readInt(0, MOD - 1);
     }
 
 

diff --git a/polynomial/sqrt_of_formal_power_series_sparse/checker.cpp b/polynomial/sqrt_of_formal_power_series_sparse/checker.cpp
@@ -2,6 +2,7 @@
 #include <iostream>
 #include <cassert>
 #include "testlib.h"
+#include "params.h"
 
 using namespace std;
 using uint = unsigned int;
@@ -115,11 +116,11 @@ bool has_sqrt(vector<Mint> pol) {
 
 bool read_ans(InStream& stream, vector<Mint> expect) {
   int n = int(expect.size());
-  int first = stream.readInt();
+  int first = stream.readInt(-1, MOD - 1);
   if (first == -1) return false;
   vector<Mint> pol(n);
   pol[0] = first;
-  for (int i = 1; i < n; i++) { pol[i] = stream.readInt(); }
+  for (int i = 1; i < n; i++) { pol[i] = stream.readInt(0, MOD - 1); }
 
   auto pol2 = multiply(pol, pol);