From 73d9ae9b72890aae4822fb0c8542faa18769a301 Mon Sep 17 00:00:00 2001
From: Barrett Ruth <br.barrettruth@gmail.com>
Date: Tue, 16 Sep 2025 00:44:13 -0400
Subject: [PATCH] final fixes

---
 lua/cp/test.lua | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lua/cp/test.lua b/lua/cp/test.lua
index 8d02ac7..c8144ff 100644
--- a/lua/cp/test.lua
+++ b/lua/cp/test.lua
@@ -170,9 +170,14 @@ local function run_single_test_case(ctx, contest_config, test_case)
 
 	local run_cmd = build_command(language_config.run, language_config.executable, substitutions)
 
+	local stdin_content = test_case.input .. "\n"
+	if ctx.contest == "atcoder" then
+		stdin_content = "1\n" .. stdin_content
+	end
+
 	local start_time = vim.uv.hrtime()
 	local result = vim.system(run_cmd, {
-		stdin = test_case.input .. "\n",
+		stdin = stdin_content,
 		timeout = contest_config.timeout_ms or 2000,
 		text = true,
 	}):wait()