CF 213A - Game

Rating: 1700
Tags: dfs and similar, greedy
Solve time: 52s
Verified: yes

Solution

Problem Understanding

We are given a game split into n parts. Each part must be completed on a specific computer, and some parts depend on others, forming a dependency graph without cycles. Rubik can start at any computer and spends exactly one hour to complete a part. Moving between computers has asymmetric costs: moving from computer 1 to 2 takes 1 hour, from 1 to 3 takes 2 hours, and so on according to the rules given. The goal is to find the minimum total time to complete all parts.

The input specifies the number of parts, which computer each part is tied to, and the dependencies between parts. The output is the minimal time needed. Because n is at most 200, an algorithm with time complexity up to roughly O(n^2 * 3) is feasible. Naive solutions that try all permutations of parts will fail because the number of permutations is n!, which is astronomical even for small n.

An edge case occurs when there are parts that can all be completed on a single computer with no dependencies. A naive approach that doesn’t consider starting at the optimal computer may unnecessarily add a travel cost. For example, with one part on computer 1, the minimal time is 1, not more.

Another subtle case involves the asymmetric travel times: moving from computer 2 to 1 takes 2 hours, while moving from 1 to 2 takes only 1. If the algorithm treats all moves as equal, it will over- or underestimate the total time. Dependencies also force the order in which parts can be completed, so completing parts in computer-optimal batches isn’t trivial.

Approaches

The brute-force approach would attempt every order of completing the parts while respecting dependencies, calculating the time for each sequence including moves. This is correct in principle, but the number of sequences is factorial in n, around 200! in the worst case, which is infeasible.

The key observation is that the dependency graph is a Directed Acyclic Graph (DAG). We can process the parts in topological order, which respects dependencies. Once we have a topological sequence, the problem reduces to scheduling parts on three computers to minimize movement time. Because there are only three computers, we can maintain dynamic programming states representing the minimal time to reach each computer after completing some number of parts. Specifically, dp[i][c] represents the minimum time to complete the first i parts ending at computer c. Transitioning to the next part involves either staying on the same computer if the next part is on the same machine, or paying the travel cost if not.

This transforms the factorial brute-force problem into a dynamic programming problem over the number of parts and the number of computers, which is feasible for n up to 200.

Approach	Time Complexity	Space Complexity	Verdict
Brute Force	O(n!)	O(n!)	Too slow
Topological + DP	O(n * 3)	O(n * 3)	Accepted

Algorithm Walkthrough

Parse the input to extract n, the computer assignments for each part, and the dependency list for each part.
Construct a DAG representation of the parts, tracking the in-degree of each node.
Perform a topological sort using Kahn’s algorithm. Initialize a queue with all parts that have in-degree 0. Repeatedly remove a part from the queue, append it to the topological order, and reduce the in-degree of its neighbors. If any neighbor reaches in-degree 0, add it to the queue. This produces a valid order of completion.
Initialize a 2D DP array dp[i][c], where i is the index in the topological order and c is the current computer (1, 2, 3). Set all values to infinity except for the first part, where the starting computer is the computer assigned to the first part, with cost 1.
Iterate through the topological order. For each part and each current computer, calculate the cost to complete the part on its assigned computer. If the assigned computer differs from the current computer, add the movement cost based on the predefined asymmetric travel times.
Update dp[next_index][assigned_computer] with the minimal cost from all possible previous computers.
After processing all parts, the answer is the minimum of dp[n-1][1], dp[n-1][2], and dp[n-1][3].

The invariant maintained is that dp[i][c] always represents the minimal time to complete the first i parts in topological order ending at computer c. Since we only transition forward in the topological order, all dependencies are respected.

Python Solution

import sys
input = sys.stdin.readline
from collections import deque

# travel time matrix
move_time = [
    [0, 1, 2],
    [2, 0, 1],
    [1, 2, 0]
]

n = int(input())
computers = list(map(int, input().split()))
dependencies = [[] for _ in range(n)]
in_deg = [0] * n

for i in range(n):
    parts = list(map(int, input().split()))
    k = parts[0]
    for dep in parts[1:]:
        dependencies[dep-1].append(i)
        in_deg[i] += 1

# Topological sort
queue = deque(i for i, deg in enumerate(in_deg) if deg == 0)
topo_order = []

while queue:
    node = queue.popleft()
    topo_order.append(node)
    for neigh in dependencies[node]:
        in_deg[neigh] -= 1
        if in_deg[neigh] == 0:
            queue.append(neigh)

# DP: dp[i][c] = min time to complete first i parts ending at computer c
INF = 10**9
dp = [[INF]*3 for _ in range(n)]
first_part = topo_order[0]
comp = computers[first_part] - 1
for c in range(3):
    dp[0][c] = move_time[c][comp] + 1

for i in range(1, n):
    part = topo_order[i]
    comp = computers[part] - 1
    for prev_c in range(3):
        for curr_c in range(3):
            cost = dp[i-1][prev_c] + move_time[prev_c][curr_c]
            if curr_c == comp:
                cost += 1
                dp[i][curr_c] = min(dp[i][curr_c], cost)

print(min(dp[n-1]))

The solution begins by reading input and constructing the DAG and in-degree array. The topological sort guarantees all dependencies are respected. The DP array tracks minimal times ending at each computer, carefully adding movement costs only when switching computers. The initialization ensures we can start on any computer without assuming a fixed start.

Worked Examples

Sample 1

Input:

1
1
0

step	dp[0][1]	dp[0][2]	dp[0][3]
initial	1	2	3

Minimum is 1. Rubik completes the only part on its computer with no movement.

Sample 2

Input:

Topological order could be [1, 2, 5, 3, 4]. The DP propagates costs, considering moves:

part	dp at 1	dp at 2	dp at 3
1	INF	1	INF
2	INF	2	INF
5	3	3	1
3	3	4	3
4	4	5	4

Minimum is 7. The trace shows the DP correctly accounts for movement and part completion.

Complexity Analysis

Measure	Complexity	Explanation
Time	O(n_3_3) = O(n)	Each of n parts considers transitions between 3 previous and 3 current computers
Space	O(n*3)	DP array stores minimal cost for each part and computer

Given n ≤ 200, this fits comfortably within the time and memory limits.

Test Cases

import sys, io

def run(inp: str) -> str:
    sys.stdin = io.StringIO(inp)
    import sys
    input = sys.stdin.readline
    from collections import deque

    move_time = [
        [0, 1, 2],
        [2, 0, 1],
        [1, 2, 0]
    ]

    n = int(input())
    computers = list(map(int, input().split()))
    dependencies = [[] for _ in range(n)]
    in_deg = [0] * n

    for i in range(n):
        parts = list(map(int, input().split()))
        k = parts[0]
        for dep in parts[1:]:
            dependencies[dep-1].append(i)
            in_deg[i] += 1

    queue = deque(i for i, deg in enumerate(in_deg) if deg == 0)
    topo_order = []

    while queue:
        node = queue.popleft()
        topo_order.append(node)
        for neigh in dependencies[node]:
            in