написать программу с графом - C++ - Ответ 4457306

#include <string>
#include <vector>
#include <fstream>
 
using namespace std;
 
int main()
{
    // read input data
    ifstream inp("input.txt");
    int n, s;
    inp >> n >> s;
    vector<vector<int> > m(n, vector<int>(n));
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            inp >> m[i][j];
        }
    }
 
    // make symmetric
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            if (m[i][j])
                m[j][i] = 1;
        }
    }
 
    // transitive closure
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            if (m[i][j]) {
                for (int k = 0; k < n; ++k) {
                    if (m[j][k])
                        m[i][k] = 1;
                }
            }
        }
    }
 
    // count friends
    int count = 0;
    for (int i = 0; i < n; ++i) {
        if (m[s - 1][i])
            count++;
    }
 
    // write result
    ofstream out("output.txt");
    out << count;
}