Cod sursa(job #529705)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 5 februarie 2011 19:23:29
Problema Componente tare conexe Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.87 kb
#include<iostream.h>
#include<fstream.h>
#include<stdlib.h>
#define N 100001
#define M 200001
typedef struct stack
{long st[N],sp;};
stack s,q;
long n,m,deg[N]={0},*g[N],k,i,j,t,ind=0,*d[N],c[N]={0},l[N],e[N],nr,a[M],b[M],u,p,r;

void push(stack &s,long x)
{s.st[++s.sp]=x;}

long pop(stack &s)
{return s.st[s.sp--];}

int apartine(stack &s,long x)
{long i;
for(i=1;i<=s.sp;i++)
if(s.st[i]==x)
      return 1;
return 0;}

void tarjan(long i,long *nr,long *ind)
{long k,j,r;
push(s,i);
c[i]=(*ind);
l[i]=(*ind)++;
for(j=0;j<deg[i];j++)
if(c[g[i][j]]==0)
       {tarjan(g[i][j],nr,ind);
       if(l[i]>l[g[i][j]])
              l[i]=l[g[i][j]];}
else
       if(apartine(s,g[i][j])==1&&l[i]>c[g[i][j]])
              l[i]=c[g[i][j]];
if(c[i]==l[i])
       {k=0;
       (*nr)++;
       d[(*nr)]=(long*)malloc((s.sp+1)*sizeof(long));
       while(s.sp!=0)
              {r=pop(s);
              if(apartine(q,r)==0)
                   if(r!=i)
                         d[(*nr)][++k]=r;
                   else
                         break;
              else
                   break;}
       if(apartine(q,i)==0)
              {push(q,i);
              d[(*nr)][++k]=i;}
       if(k==0)
              {free(d[*nr]);
              (*nr)--;}
       else
              e[(*nr)]=k;}}

int main()
{ifstream f1("ctc.in");
ofstream f2("ctc.out");
f1>>n>>m;
for(k=1;k<=m;k++)
       {f1>>a[k]>>b[k];
       deg[a[k]]++;}
for(i=1;i<=n;deg[i++]=0)
       g[i]=(long*)malloc(deg[i]*sizeof(long));
for(j=1;j<=m;j++)
       g[a[j]][deg[a[j]]++]=b[j];
ind=0;
nr=0;
s.sp=0;
q.sp=0;
for(u=1;u<=n;u++)
if(c[u]==0)
       tarjan(u,&nr,&ind);
for(t=1;t<=n;t++)
       free(g[t]);
f2<<nr<<endl;
for(p=1;p<=nr;p++)
       {for(r=1;r<=e[p];r++)
              f2<<d[p][r]<<" ";
       free(d[p]);
       f2<<endl;}
f1.close();
f2.close();
return 0;}