Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
DFS from source, no extra data types. solution in Clear category for Node Disconnected Users by JamesArruda
def visible_nodes(curr, grid, off, viz):
for a, b in grid:
if a == curr and b not in off:
viz.add(b)
visible_nodes(b, grid, off, viz)
def disconnected_users(net, users, source, crushes):
viz = set(source)
if source in crushes:
return sum(users.values())
visible_nodes(source, net, crushes, viz)
return sum(users[x] for x in users if x not in viz)
if __name__ == '__main__':
#These "asserts" using only for self-checking and not necessary for auto-testing
assert disconnected_users([
['A', 'B'],
['B', 'C'],
['C', 'D']
], {
'A': 10,
'B': 20,
'C': 30,
'D': 40
},
'A', ['C']) == 70, "First"
assert disconnected_users([
['A', 'B'],
['B', 'D'],
['A', 'C'],
['C', 'D']
], {
'A': 10,
'B': 0,
'C': 0,
'D': 40
},
'A', ['B']) == 0, "Second"
assert disconnected_users([
['A', 'B'],
['A', 'C'],
['A', 'D'],
['A', 'E'],
['A', 'F']
], {
'A': 10,
'B': 10,
'C': 10,
'D': 10,
'E': 10,
'F': 10
},
'C', ['A']) == 50, "Third"
print('Done. Try to check now. There are a lot of other tests')
Sept. 14, 2018
