LeetCode in Kotlin

3812. Minimum Edge Toggles on a Tree

Hard

You are given an undirected tree with n nodes, numbered from 0 to n - 1. It is represented by a 2D integer array edges of length n - 1, where edges[i] = [a_i, b_i] indicates that there is an edge between nodes a_i and b_i in the tree.

You are also given two binary strings start and target of length n. For each node x, start[x] is its initial color and target[x] is its desired color.

In one operation, you may pick an edge with index i and toggle both of its endpoints. That is, if the edge is [u, v], then the colors of nodes u and v each flip from '0' to '1' or from '1' to '0'.

Return an array of edge indices whose operations transform start into target. Among all valid sequences with minimum possible length, return the edge indices in increasing order.

If it is impossible to transform start into target, return an array containing a single element equal to -1.

Example 1:

Input: n = 3, edges = [[0,1],[1,2]], start = “010”, target = “100”

Output: [0]

Explanation:

Toggle edge with index 0, which flips nodes 0 and 1.
The string changes from "010" to "100", matching the target.

Example 2:

Input: n = 7, edges = [[0,1],[1,2],[2,3],[3,4],[3,5],[1,6]], start = “0011000”, target = “0010001”

Output: [1,2,5]

Explanation:

Toggle edge with index 1, which flips nodes 1 and 2.
Toggle edge with index 2, which flips nodes 2 and 3.
Toggle edge with index 5, which flips nodes 1 and 6.

After these operations, the resulting string becomes "0010001", which matches the target.

Example 3:

Input: n = 2, edges = [[0,1]], start = “00”, target = “01”

Output: [-1]

Explanation:

There is no sequence of edge toggles that transforms "00" into "01". Therefore, we return [-1].

Constraints:

2 <= n == start.length == target.length <= 10⁵
edges.length == n - 1
edges[i] = [a_i, b_i]
0 <= a_i, b_i < n
start[i] is either '0' or '1'.
target[i] is either '0' or '1'.
The input is generated such that edges represents a valid tree.

Solution

class Solution {
    private lateinit var visited: BooleanArray
    private lateinit var adjList: Array<ArrayList<IntArray>>
    private lateinit var flips: IntArray
    private lateinit var ans: MutableList<Int>
    private lateinit var s: String
    private lateinit var t: String

    fun minimumFlips(n: Int, edges: Array<IntArray>, start: String, target: String): MutableList<Int> {
        s = start
        t = target
        flips = IntArray(n)
        adjList = Array(n) { ArrayList<IntArray>() }
        for (i in edges.indices) {
            val edge = edges[i]
            adjList[edge[0]].add(intArrayOf(edge[1], i))
            adjList[edge[1]].add(intArrayOf(edge[0], i))
        }
        ans = ArrayList()
        visited = BooleanArray(n)
        dfs(0, -1, -1)
        // Sort indices as required
        val sortedAns = ans.sorted().toMutableList()
        // If root is still mismatched, it's impossible (no parent to flip)
        if (mismatch(0)) {
            sortedAns.clear()
            sortedAns.add(-1)
        }
        return sortedAns
    }

    // Helper to check if current state matches target
    // Logic: If flips is even, bit is original. If flips is odd, bit is inverted.
    private fun mismatch(u: Int): Boolean = ((flips[u] % 2 == 0) && s[u] != t[u]) ||
        ((flips[u] % 2 == 1) && s[u] == t[u])

    private fun dfs(u: Int, parent: Int, index: Int) {
        visited[u] = true
        for (v in adjList[u]) {
            if (!visited[v[0]]) {
                dfs(v[0], u, v[1])
            }
        }
        if (parent != -1 && mismatch(u)) {
            ans.add(index)
            flips[u]++
            flips[parent]++
        }
    }
}

This site is open source. Improve this page.