Advertisement
Guest User

Untitled

a guest
Jan 28th, 2015
175
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.51 KB | None | 0 0
  1. n, s = list(map(int, input().split()))
  2. graph = [[] for i in range(n)]
  3. for i in range(n):
  4.     arr = list(map(int, input().split()))
  5.     for j in range(n):
  6.         if arr[j] == 1:
  7.             graph[i].append(j)
  8.  
  9. ans = 1
  10. used = [False for i in range(n)]
  11. used[s - 1] = True
  12.  
  13. def dfs(s):
  14.     global ans, used, graph
  15.     for i in range(len(graph[s])):
  16.         if used[graph[s][i]] == False:
  17.             used[graph[s][i]] = True
  18.             ans += 1
  19.             dfs(graph[s][i])
  20.     return
  21.  
  22. dfs(s - 1)
  23. print(ans)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement