Cod sursa(job #2386167)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 22 martie 2019 12:25:04
Problema Andrei Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.98 kb
#include<bits/stdc++.h>
using namespace std;
ifstream f("andrei.in");
ofstream g("andrei.out");
int n, m, color[100002];
vector<pair<int, int> >v[100002];
bool viz[100002];
void dfs(int nod, int col)
{
    viz[nod] = 1;
    color[nod] = col;
    for(int i = 0; i < v[nod].size(); ++i)
    {
        int vecin = v[nod][i].first;
        int cc = v[nod][i].second;
        if(!viz[vecin] && cc == col)
            dfs(vecin, col ^ 1);
        else
            if(!viz[vecin] && cc == 2)
                dfs(vecin, col);
            else
                if(!viz[vecin] && cc != col)
                    dfs(vecin, col);
    }
}
int main()
{
    f >> n >> m;
    for(int i = 1; i <= m; ++i)
    {
        int a, b, c;
        f >> a >> b >> c;
        v[a].push_back({b, c});
        v[b].push_back({a, c});
    }
    for(int i = 1; i <= n; ++i)
        if(!viz[i])
            dfs(i, 0);
    for(int i = 1; i <= n; ++i)
        g << color[i] << " ";
    return 0;
}