testlib_for_yukicoder.cpp

#include <iostream>
#include "testlib.h"


std::ifstream source_ifs;
std::ofstream score_ofs;

#ifndef IS_NOT_YUKICODER
/*
 * ./a.out <input-file> <answer-file> <source-file> <score-file> (+ output from stdin)
 */
void registerTestlibForYukicoder(int argc, char* argv[]) {
    __testlib_ensuresPreconditions();

    testlibMode = _checker;
    __testlib_set_binary(stdin);

    if (argc > 1 && !strcmp("--help", argv[1]))
        __testlib_help();

    if (argc != 5) {
        quit(_fail, std::string("Program must be run with the following arguments: ") +
            std::string("<input-file> <answer-file> <source-file> <score-file> (+ output from stdin)"));
    }
    resultName = "";
    appesMode = false;

    inf.init(argv[1], _input);
    ans.init(argv[2], _answer);
    ouf.init(stdin, _output);

    source_ifs.open(argv[3], std::ios::in);
    if (source_ifs.fail() || !source_ifs.is_open())
        quit(_fail, std::string("Can not read the source-file '") + argv[3] + std::string("'"));

    score_ofs.open(argv[4], std::ios::out);
    if (score_ofs.fail() || !score_ofs.is_open())
        quit(_fail, std::string("Can not write to the score-file '") + argv[4] + std::string("'"));
}
#endif



bool is_prime(long long n) {
    if (n < 2) return false;
    for (long long i = 2; i * i <= n; ++i) {
        if (n % i == 0) {
            return false;
        }
    }
    return true;
}

int main(int argc, char** argv) {

#ifdef IS_NOT_YUKICODER
    // You have to edit here if you use the reactive or scoring judge.
    registerTestlibCmd(argc, argv);
#else
    registerTestlibForYukicoder(argc, argv);
#endif

    long long n = inf.readLong();
    long long p = ouf.readLong(2, n);
    if (!is_prime(p)) {
        quitf(_wa, "%lld is not prime", p);
    }
    if (!ouf.seekEof()) {
        quitf(_wa, "Participant output contains extra tokens");
    }
    quitf(_ok, "ok");
    return 0;
}

~readme.md

手元で Rime を用いてテストする場合は、TESTSET で

cxx_judge(src='judge.cc', dependency=['testlib.h'], flags=['-DIS_NOT_YUKICODER'], variant=testlib_judge_runner)

みたいな感じに flag を指定すると動くと思います。

リアクティブ、スコア問題に関しては全くテストしていないので動く保証はないです。