Maximize Spanning Tree Stability with Upgrades

You are given an integer n, representing n nodes numbered from 0 to n - 1, and a list of edges, where edges[i] = [ui, vi, si, musti]:

  • ui and vi indicate an undirected edge between nodes ui and vi.
  • si is the strength of the edge.
  • musti is an integer (0 or 1). If musti == 1, the edge must be included in the spanning tree. These edges cannot be upgraded.

You are also given an integer k, the maximum number of upgrades you can perform. Each upgrade doubles the strength of an edge, and each eligible edge with musti == 0 can be upgraded at most once.

The stability of a spanning tree is defined as the minimum strength score among all edges included in it.

Return the maximum possible stability of any valid spanning tree. If it is impossible to connect all nodes, return -1.

Note: A spanning tree of a graph with n nodes is a subset of the edges that connects all nodes together, meaning the graph is connected, without forming any cycles, and uses exactly n - 1 edges.

Example 1
Inputn = 3, edges = [[0,1,2,1],[1,2,3,0]], k = 1
Output2
Edge [0,1] with strength 2 must be included, edge [1,2] can be upgraded from 3 to 6, and the minimum strength in the resulting spanning tree is 2.
Example 2
Inputn = 3, edges = [[0,1,4,0],[1,2,3,0],[0,2,1,0]], k = 2
Output6
Upgrade edges [0,1] from 4 to 8 and [1,2] from 3 to 6, so the minimum strength in the resulting spanning tree is 6.

Constraints

  • 2 <= n <= 10^5
  • 1 <= edges.length <= 10^5
  • edges[i] = [ui, vi, si, musti]
  • 0 <= ui, vi < n
  • ui != vi
  • 1 <= si <= 10^5
  • musti is either 0 or 1.
  • 0 <= k <= n
  • There are no duplicate edges.

Asked at 3 companies

</>

Your Solution

(Ctrl/Cmd + Enter)

Switching Language

Loading template...

Loading...

Sign in to save your progress

AI code evaluation

Get a correctness verdict, missed edge cases, and complexity analysis of your solution.

Sign in to evaluate