First clear solution for How to Find Friends by agdk26 - python coding challenges

Enable Javascript in your browser and then refresh this page, for a much enhanced experience.

First solution in Clear category for How to Find Friends by agdk26

def check_connection(network, first, second):
    friends=set() #connected already seen
    stack=set() #connected to see
    stack.add(first)
    while stack:
        droid=stack.pop()
        friends.add(droid)
        if droid==second: return True #found
        for neighbour in get_neighbours(network,droid):
            if neighbour not in friends:
                stack.add(neighbour)
    return False
    
def get_neighbours(network, droid):
    for connection in network:
        left,right=connection.split('-')
        if droid==left: yield right
        if droid==right: yield left

if __name__ == '__main__':
    #These "asserts" using only for self-checking and not necessary for auto-testing
    assert check_connection(
        ("dr101-mr99", "mr99-out00", "dr101-out00", "scout1-scout2",
         "scout3-scout1", "scout1-scout4", "scout4-sscout", "sscout-super"),
        "scout2", "scout3") == True, "Scout Brotherhood"
    assert check_connection(
        ("dr101-mr99", "mr99-out00", "dr101-out00", "scout1-scout2",
         "scout3-scout1", "scout1-scout4", "scout4-sscout", "sscout-super"),
        "super", "scout2") == True, "Super Scout"
    assert check_connection(
        ("dr101-mr99", "mr99-out00", "dr101-out00", "scout1-scout2",
         "scout3-scout1", "scout1-scout4", "scout4-sscout", "sscout-super"),
        "dr101", "sscout") == False, "I don't know any scouts."

June 5, 2014

Comments:

artemrudenko on Oct. 2, 2014, 5:49 a.m.
Nice and short

snowshadow on Nov. 14, 2014, 7:50 a.m.
brilliant

Eldin on Dec. 2, 2014, 3:11 p.m.
Nice work using the set and generator!

rofffo on Sept. 19, 2015, 9:23 p.m.
Very clean and light, yet powerful. I like this solution very much and this kind of solutions in general. It looks like you spent some time thinking about how to model the situation in some clear and insightful way and then described the model in code, instead of just describing the situation directly. I know how to learn to write code, not so much sure about how to learn to see these elegant models. :)

Stensen on Dec. 20, 2020, 11:57 a.m.
That's intelligent!

frog_frog00 on May 10, 2021, 10:38 a.m.
Nice

kxy_2333 on May 8, 2022, 11:31 a.m.
Good.I have spent points to unlock that riddle,because I couldn´t find out how to go into this.Your code is beautifull.

Nocturne13 on May 5, 2023, 9:03 a.m.
Very nice example of using utility function for solving this task. I suppose the benefit of using yield statement instead of return in get_neighbours function is just better time efficiency?

GitHub

0 %

You ClassRoom Demo is Ready With the new tool CheckiO ClassRoom, you will be able to analyze your students' progress, have your own leaderboard, change missions order on the map, and so on.