-
Notifications
You must be signed in to change notification settings - Fork 0
/
BFS.cpp
41 lines (38 loc) · 873 Bytes
/
BFS.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
//Distance stores levels, and ways stores the number of ways to get from kth vertex to any other vertex with the shortest path taken
#include <bits/stdc++.h>
using namespace std;
int vis[N], dist[N], ways[N];
vector<int> g[N];
void bfs(int k)
{
queue<int> q;
q.push(k);
ways[k] = 1;
vis[k] = 1;
while (!q.empty())
{
int node = q.front();
q.pop();
for (auto it : g[node])
{
if (!vis[it])
{
dist[it] = dist[node] + 1;
vis[it] = 1;
ways[it] += ways[node];
q.push(it);
}
else
{
if (dist[node] + 1 == dist[it])
{
ways[it] += ways[node];
}
}
}
}
}
//Problem 1: https://codeforces.com/contest/208/problem/C
//Solution 1: https://codeforces.com/contest/208/submission/38315030
//Problem 2 (Multiple of number not containing some digits): https://www.spoj.com/problems/MULTII/
//Solution 2: http://p.ip.fi/li3d