Не могу сдать задачу на dfs на informatics.msk.ru
Условие задачи отсюда https://informatics.msk.ru/mod/statements/view.php?id=256#1. Мой исходный код:
#include <stack>
#include "iostream"
#include "queue"
using namespace std;
void dfs(int v, vector<vector<int>> a, vector<bool>& used, int n, int& count) {
used[v] = 1;
count++;
for (int i = 0; i < n; i++) {
if (a[v][i] != 0 && !used[i]) {
dfs(a[v][i], a, used, n, count);
}
}
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
int n, s;
cin >> n >> s;
s--;
vector<vector<int>> a(n, vector<int>(n));
for (auto i = 0; i < n; i++) {
for (auto j = 0; j < n; j++) {
cin >> a[i][j];
}
}
vector<bool> used(n, 0);
int count = 0;
dfs(s, a, used, n, count);
cout << count;
}
Мой компьютер запускает алгоритм для данных из первого тестового случая корректно. В чем проблема?
