From 7c0bdf3d8d298145eeaf7a19666d2094e15f1353 Mon Sep 17 00:00:00 2001
From: Baoshuo <i@baoshuo.ren>
Date: Wed, 5 Oct 2022 18:53:34 +0800
Subject: [PATCH] refactor: yesno checker

---
 judger/uoj_judger/builtin/checker/yesno.cpp | 50 +++++++++++++++------
 1 file changed, 37 insertions(+), 13 deletions(-)

diff --git a/judger/uoj_judger/builtin/checker/yesno.cpp b/judger/uoj_judger/builtin/checker/yesno.cpp
index ea605c8..b8adb21 100644
--- a/judger/uoj_judger/builtin/checker/yesno.cpp
+++ b/judger/uoj_judger/builtin/checker/yesno.cpp
@@ -1,27 +1,51 @@
 #include "testlib.h"
-#include <string>
 
 using namespace std;
 
 const string YES = "YES";
 const string NO = "NO";
 
-int main(int argc, char * argv[])
-{
-    setName("YES or NO (case insensetive)");
+int main(int argc, char* argv[]) {
+    setName("compare ordered sequences of YES or NO (case insensetive)");
+
     registerTestlibCmd(argc, argv);
 
-    std::string ja = upperCase(ans.readWord());
-    std::string pa = upperCase(ouf.readWord());
+    int n = 0;
 
-    if (ja != YES && ja != NO)
-        quitf(_fail, "%s or %s expected in answer, but %s found", YES.c_str(), NO.c_str(), compress(ja).c_str());
+    while (!ans.seekEof() && !ouf.seekEof()) {
+        n++;
+        std::string ja = upperCase(ans.readWord());
+        std::string pa = upperCase(ouf.readWord());
 
-    if (pa != YES && pa != NO)
-        quitf(_pe, "%s or %s expected, but %s found", YES.c_str(), NO.c_str(), compress(pa).c_str());
+        if (ja != YES && ja != NO)
+            quitf(_fail, "%d%s differ - %s or %s expected, but %s found", n, englishEnding(n).c_str(), YES.c_str(), NO.c_str(), compress(ja).c_str());
 
-    if (ja != pa)
-        quitf(_wa, "expected %s, found %s", compress(ja).c_str(), compress(pa).c_str());
+        if (pa != YES && pa != NO)
+            quitf(_pe, "%d%s differ - %s or %s expected, but %s found", n, englishEnding(n).c_str(), YES.c_str(), NO.c_str(), compress(pa).c_str());
 
-    quitf(_ok, "answer is %s", ja.c_str());
+        if (ja != pa)
+            quitf(_wa, "%d%s differ - expected: '%s', found: '%s'", n, englishEnding(n).c_str(), vtos(ja).c_str(), vtos(pa).c_str());
+    }
+
+    int extraInAnsCount = 0;
+
+    while (!ans.seekEof()) {
+        ans.readToken();
+        extraInAnsCount++;
+    }
+
+    int extraInOufCount = 0;
+
+    while (!ouf.seekEof()) {
+        ouf.readToken();
+        extraInOufCount++;
+    }
+
+    if (extraInAnsCount > 0)
+        quitf(_wa, "Answer contains longer sequence [length = %d], but output contains %d elements", n + extraInAnsCount, n);
+
+    if (extraInOufCount > 0)
+        quitf(_wa, "Output contains longer sequence [length = %d], but answer contains %d elements", n + extraInOufCount, n);
+
+    quitf(_ok, "%d token(s)", n);
 }