IITWPC4I - Petya and the Road Repairs
Petya is the mayor of a city named Mayapur. In the morning, everybody likes to drink hot tea in bed, and the citizens need milk to take with their tea. For this purpose, they should be able to reach at least some milkman in the city. There are m bi-directional roads in the city but all of them are currently unrepaired, hence not in a state of use.
Petya cares about his city a lot and intends to repair some of these roads, so that every citizen is connected to at least one milkman. For repairing each road he needs to pay a cost. He would like to minimize the cost of this project. Note that a milkman does not need to go to some other milkman for milk as he can take milk from himsaelf.
Help Petya in finding out the minimum cost needed to repair the roads in the given way. If it is not possible for a citizen to connect to any of the milkmen, output "impossible" (without quotes).
The first line contains T: the number of test cases. (1 <= T <= 100)
For each test case, the first line contains two space-seperated numbers n, m: n is the number of citizens in Mayapur and m denotes the number of unrepaired roads (1 <= n <= 10^5, 1 <= m <= min (n * (n - 1) / 2, 2 * 10^5)).
The next line contains n space-separated integers, which are either 0 or 1, denoting for successive citizens whether they are a milkman or not.
Then, for each of the next m lines, each line contains three space-seperated integers u, v and c, denoting that there exists a unrepaired road between u and v such that the cost of repairing of road is c. (1 <= u, v <= n and u != v, 1 <= c <= 10^9.)
For each test case output the cost as required.
Input: 1 5 7 0 1 0 1 0 1 2 11 1 3 1 1 5 17 2 3 1 3 5 18 4 5 3 2 4 5 Output: 5
Ignoring "impossible" cost me 2 WA.
multisource prim's algo
In java using array to store edges and sorting them gave TLE while using priority queue it gets an ACLast edit: 2020-06-10 04:26:10
Can also be solved by Dijkstra's algo
Please note: If citizen is a milkman then value is 1 else it is zero.
A very good question for people who want to implement Kruskal's. Don't know about Prims (lol) and not reading the question costed me 2WA.
everyone is milkman:):)
Make arrays till 3*10^5 .....costed me several RTE .... Nice problem ......
there can be multiple roads b/w 2 cities.
DSU; do not connect already connected nodes!