Cod sursa(job #1894762)

Utilizator Bodo171Bogdan Pop Bodo171 Data 27 februarie 2017 14:56:34
Problema Andrei Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
const int nmax=200005;
vector<int> v[nmax],vt[nmax];
bool viz[nmax],val[nmax];
int p[nmax];
int n,x,y,tip,k,m,i,a,b;
inline int no(int x)
{
    return (x+(2*n)*(x<=n)-n);
}
void add(int x,int y)
{
    v[no(x)].push_back(y);
    vt[y].push_back(no(x));
    v[no(y)].push_back(x);
    vt[x].push_back(no(y));
}
void dfs(int x)
{
    viz[x]=1;
    for(int i=0;i<v[x].size();i++)
        if(!viz[v[x][i]])
         dfs(v[x][i]);
    k++;p[k]=x;
}
void dfst(int x)
{
    viz[x]=0;val[x]=0;val[no(x)]=1;
    for(int i=0;i<vt[x].size();i++)
        if(viz[vt[x][i]])
          dfst(vt[x][i]);
}
int main()
{
    ifstream f("andrei.in");
    ofstream g("andrei.out");
    f>>n>>m;
    //A-0,B-1
    for(i=1;i<=m;i++)
    {
        f>>a>>b>>tip;
        if(tip==0)//alb
        {
            add(a,b);
        }
        if(tip==1)//rosu
        {
            add(no(a),no(b));
        }
        if(tip==2)
        {
            add(a,no(b));
            add(no(a),b);
        }
    }
    for(i=1;i<=2*n;i++)
        if(!viz[i])
        dfs(i);
    for(i=2*n;i>=1;i--)
    {
        x=p[i];
        if(viz[x]&&viz[no(x)])
            dfst(x);
    }
    for(i=1;i<=n;i++)
        g<<val[i]<<' ';
    return 0;
}