Uvalive 6264 Conservation 拓扑排序

题意:有两个实验室,有n个实验和m对实验依赖 (a,b) 代表实验 b 必须在实验 a 之后才能进行,一开始可以在任意两个实验室进行实验,问最少要转移多少次实验室可以把这n个实验全部完成。
两次拓扑排序分别计算从1号实验室开始和从2号实验室开始所需要转移的次数就可以。。。
代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int MAXN = 100000+1;
int n,m,T;
struct Edge {
    int to,next;
    Edge(){}
    Edge(int _to,int _next):to(_to),next(_next){}
}e[1000000+5];
int head[100000+1],tot;
void init(){
for(int i=1;i<=n;i++)head[i]=-1;
// memset(head,-1,sizeof(head));
    tot=0;
}
void AddEdge(int u,int v){
    e[tot]=Edge(v,head[u]);
    head[u]=tot++;
}
void Read(int &x){
    char ch;
    while((ch=getchar())&&!isdigit(ch));
    x=0;
    do {
        x=(x<<1)+(x<<3)+(ch^48);
    }while((ch=getchar())&&isdigit(ch));
}
int que1[MAXN*2],que2[MAXN*2];
int front1,rear1,front2,rear2;
int du[MAXN],type[MAXN],du1[MAXN];

int countt(){
    front1=rear1=front2=rear2=0;
    //memcpy(du1,du,sizeof(du));
    for(int i=1;i<=n;i++){
        du1[i]=du[i];
        if(du[i]==0){
            if(type[i]==1) que1[rear1++]=i;
            if(type[i]==2) que2[rear2++]=i;
            --du[i];
        }
    }
    int cnt=0;
    int pre=1;
    //if(rear1==0)pre=2;
    //else pre=1;
    while((front1!=rear1)||(front2!=rear2)){
        int tf=0;
        while(front1!=rear1){
            tf=1;
            int u=que1[front1++];
            for(int i=head[u];i!=-1;i=e[i].next){
                int v=e[i].to;
                du[v]--;
                if(du[v]==0){
                    if(type[v]==1)que1[rear1++]=v;//q1.push(v);
                    if(type[v]==2)que2[rear2++]=v;//q2.push(v);
                    du[v]--;
                }
            }
        }
        if(tf&&pre!=1){
            pre=1;++cnt;
        }
        tf=0;
        while(front2!=rear2){
            tf=1;
            int u=que2[front2++];
            for(int i=head[u];i!=-1;i=e[i].next){
                int v=e[i].to;
                du[v]--;
                if(du[v]==0){
                    if(type[v]==1)que1[rear1++]=v;//q1.push(v);
                    if(type[v]==2)que2[rear2++]=v;//q2.push(v);
                    du[v]--;
                }
            }
        }
        if(tf&&pre!=2){
            pre=2;++cnt;
        }
    }

 // memcpy(du,du1,sizeof(du));
    front1=rear1=front2=rear2=0;
    for(int i=1;i<=n;i++){
        du[i]=du1[i];
        if(du[i]==0){
            if(type[i]==1)que1[rear1++]=i;//q1.push(i);
            if(type[i]==2)que2[rear2++]=i;//q2.push(i);
            --du[i];
        }
    }
    int cnt1=0;
    pre=2;
    while((front1!=rear1)||(front2!=rear2)){
        int tf=0;
        while(front2!=rear2){
            tf=1;
            int u=que2[front2++];
            for(int i=head[u];i!=-1;i=e[i].next){
                int v=e[i].to;
                du[v]--;
                if(du[v]==0){
                    if(type[v]==1)que1[rear1++]=v;//q1.push(v);
                    if(type[v]==2)que2[rear2++]=v;//q2.push(v);
                    du[v]--;
                }
            }
        }
        if(tf&&pre!=2){
            pre=2;++cnt1;
        }
        tf=0;
        while(front1!=rear1){
            tf=1;
            int u=que1[front1++];
            for(int i=head[u];i!=-1;i=e[i].next){
                int v=e[i].to;
                du[v]--;
                if(du[v]==0){
                    if(type[v]==1)que1[rear1++]=v;//q1.push(v);
                    if(type[v]==2)que2[rear2++]=v;//q2.push(v);
                    du[v]--;
                }
            }
        }
        if(tf&&pre!=1){
            pre=1;++cnt1;
        }
    }
    return cnt<cnt1?cnt:cnt1;

    //return cnt;
}
int main()
{
    //while(~scanf("%d",&T))
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        init();memset(du,0,sizeof(du));
        for(int i=1;i<=n;i++) Read(type[i]);//scanf("%d",&type[i]);
        for(int i=0,x,y;i<m;i++){
           // scanf("%d%d",&x,&y);
            Read(x),Read(y);
            AddEdge(x,y);

            ++du[y];
        }
        printf("%d\n",countt());
    }
    return 0;
}

你可能感兴趣的:(拓扑排序)