Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <iostream>
- #include <string>
- #include <sstream>
- #include <cmath>
- #include <algorithm>
- #include <memory.h>
- #include <stdio.h>
- #include <vector>
- #include <stack>
- #include <deque>
- #include <queue>
- #include <vector>
- #include <set>
- #include <iterator>
- #include <map>
- #define int long long
- #define sp system("pause")
- #define pb push_back
- #define endl '\n'
- using namespace std;
- int mas[100][100]; // матрица смежности
- int mark[10000]; // массив с отметками, посещали мы вершину или нет
- vector <set<int>> gr(10000); // список смежности
- int ans = 0;
- // функция перевода матрицы смежности в список смежности
- void in_set(int n)
- {
- for (int i = 0; i < n; i++)
- {
- for (int j = 0; j < n; j++)
- {
- if (mas[i][j] == 1)
- {
- gr[i].insert(j);
- }
- }
- }
- return;
- }
- // обход в глубину
- void DFS(int v) // v - вершина, из которой рассматриваем дальнейшие вершины
- {
- mark[v] = 1; // помечаем вершину, что мы в ней были
- for (int i : gr[v]) // идем по всем вершинам, куда можно попасть из v
- {
- if (mark[i] == 0) // если мы в вершине не были, то запускаем функцию оттуда
- {
- DFS(i);
- ans++;
- }
- }
- return;
- }
- signed main()
- {
- ios_base::sync_with_stdio(0);
- cin.tie(0);
- cout.tie(0);
- int n, s;
- cin >> n >> s;
- // ввод матрицы смежности
- for (int i = 0; i < n; i++)
- {
- for (int j = 0; j < n; j++)
- {
- cin >> mas[i][j];
- }
- }
- // перевод матрицы в список смежности
- in_set(n);
- // s -- вершина, из которой мы запускаем дфс изначально
- s--;
- DFS(s);
- cout << ans;
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement