Terminal-Bench

120

usaco==head

Usage

uv tool install terminal-bench

pip install terminal-bench

tb run \
  --dataset usaco==head \
  --agent terminus \
  --model anthropic/claude-sonnet-4-20250514 \
  --task-id 120

New to terminal-bench? See our quickstart guide.

Instruction

Please implement a Python 3 solution to the below problem. Reason through the problem and: 1. Restate the problem in plain English 2. Conceptualize a solution first in plain English 3. Write a pseudocode solution 4. Save your solution as solution.py No outside libraries are allowed. [BEGIN PROBLEM] Problem 3: Nearby Cows [Neal Wu and Eric Price, 2011] Farmer John has noticed that his cows often move between nearby fields. Taking this into account, he wants to plant enough grass in each of his fields not only for the cows situated initially in that field, but also for cows visiting from nearby fields. Specifically, FJ's farm consists of N fields (1 <= N <= 100,000), where some pairs of fields are connected with bi-directional trails (N-1 of them in total). FJ has designed the farm so that between any two fields i and j, there is a unique path made up of trails connecting between i and j. Field i is home to C(i) cows, although cows sometimes move to a different field by crossing up to K trails (1 <= K <= 20). FJ wants to plant enough grass in each field i to feed the maximum number of cows, M(i), that could possibly end up in that field -- that is, the number of cows that can potentially reach field i by following at most K trails. Given the structure of FJ's farm and the value of C(i) for each field i, please help FJ compute M(i) for every field i. PROBLEM NAME: nearcows INPUT FORMAT: * Line 1: Two space-separated integers, N and K. * Lines 2..N: Each line contains two space-separated integers, i and j (1 <= i,j <= N) indicating that fields i and j are directly connected by a trail. * Lines N+1..2N: Line N+i contains the integer C(i). (0 <= C(i) <= 1000) SAMPLE INPUT: 6 2 5 1 3 6 2 4 2 1 3 2 1 2 3 4 5 6 INPUT DETAILS: There are 6 fields, with trails connecting (5,1), (3,6), (2,4), (2,1), and (3,2). Field i has C(i) = i cows. OUTPUT FORMAT: * Lines 1..N: Line i should contain the value of M(i). SAMPLE OUTPUT: 15 21 16 10 8 11 OUTPUT DETAILS: Field 1 has M(1) = 15 cows within a distance of 2 trails, etc. [END PROBLEM]

120

Usage

Instruction

Tags