LDUOJ周练--保存名画--拓扑排序

Description
在一个Byteland上有一副非常有名的图画需要被封存。这个作品需要在2个实验室进行处理。这个处理过程被分为许多步骤。对于每个步骤，我们知道它必须要在哪个实验室进行。
在两个实验室之间运输这些这些美丽但又易碎的画会带来额外的风险，因此这个运输的过程需要尽可能的被避免。理想情况下所有的工作都会在第一个实验室做完，然后剩下的在另一个实验室完成，但是有些步骤必须在另一步骤完成之后才能完成。你的任务是找到一个工作的顺序，让这个被运输的过程最少。输出需要运输的次数。

Input
第一行N和M，代表着N个步骤，和M个先后关系。下面一行N个数字，第I个数字是1或2，代表了第I个工作需要在哪个实验室完成，下面M行I,J，代表了第I个工作必须在第J个工作前完成。
你可以认为这个题目必有解

Output
输出最少需要的运输次数。

Samples
Input Copy
5 6
1 2 1 2 1
1 2
1 3
2 4
3 4
2 5
3 5
Output
2
Hint
【数据范围与约定】

30%数据保证n≤5。
100%数据中n≤100000，m≤1000000。

题目大意

n个点，每个点有个一值（1或者2），m条边，如果边为u->v 那么必须先完成工作u再完成工作v，求一个序列使得序列中1和2的交替次数最小

题解：

m条边，如果边为u->v 那么必须先完成工作u再完成工作v

很明显是个拓扑排序了
我们要尽可能少的在序列中出现12交替，那么如果此时我们拓扑排序时出队的是u点，u点所连接的点v的入度减，如果v点成为满足条件的点而且v点与u点的点权相同的话，那么v点入队，如果不同就把v点存起来，直到队列中不存在和当前点权相同的点，我们此时不得不交换一次序列使ans++，然后重复上述操作

具体操作：

我一开始设了两个队列，一个存点权为1的点另一个存点权为2的点，然后调bug未遂（逃），
CoolGuang学长教我 用一个超级原点，连向所有的点，因为起点可能是0或者1,两次取min就ok了

10分钟后我又来更新了，我找出bugle

Code1

#include<iostream>
#include<algorithm>
#include<math.h>
#include<queue>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <vector>


using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
typedef unsigned long long ull;
const int inf = 0x3f3f3f3f;
const int maxn = 1e5 + 7;
const ll mod = 1000000007;

#define mst(x, a) memset( x,a,sizeof(x) )
#define rep(i, a, b) for(int i=(a);i<=(b);++i)
#define dep(i, a, b) for(int i=(a);i>=(b);--i)


int n,m,a[100002],d[100002],dd[100002],cnt;
vector<int>e[maxn],g;
int slove(int st) {
    int ans=-1;
    rep(i,1,n) d[i]=dd[i];
    queue<int>q;
    q.push(0);
    a[0]=st;
    int  flag=st;
    while(1) {
        ans++;
        for(int v:g) q.push(v);
        g.clear();
        while(q.size()) {
            int u=q.front();
            q.pop();
            for(int v:e[u]) {
                d[v]--;
                if(d[v]==0&&a[v]==flag) q.push(v);
                else if(d[v]==0)g.push_back(v);
            }
        }
        if(flag==1) flag=2;
        else flag=1;
        if(g.size()==0) break;
    }
    return ans;
}


int main() {
    scanf("%d%d",&n,&m);
    rep(i,1,n ) {
        scanf("%d",&a[i]);
        dd[i]++;
        e[0].push_back(i);
    }
    for(int i=1 ; i<=m ; i++) {
        int u,v;
        scanf("%d%d",&u,&v);
        dd[v]++;
        e[u].push_back(v);
    }
    int ans1=slove(1);
    int ans2=slove(2);
    cout<<min(ans1,ans2)<<endl;
    return 0;
}
/**

**/

Code2

#include<iostream>
#include<algorithm>
#include<math.h>
#include<queue>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <vector>


using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
typedef unsigned long long ull;
const int inf = 0x3f3f3f3f;
const int maxn = 2e5 + 7;
const ll mod = 1000000007;

#define mst(x, a) memset( x,a,sizeof(x) )
#define rep(i, a, b) for(int i=(a);i<=(b);++i)
#define dep(i, a, b) for(int i=(a);i>=(b);--i)

int  read() {
    int  x = 0;
    char ch = getchar();
    while(ch < '0' || ch > '9')ch = getchar();
    while(ch >= '0' && ch <= '9')x = (x << 1) + (x << 3) + (ch ^ 48), ch = getchar();
    return x;
}

void out(ll x) {
    int stackk[40];
    if (x < 0) {
        putchar('-');
        x = -x;
    }
    if (!x) {
        putchar('0');
        return;
    }
    int top = 0;
    while (x) stackk[++top] = x % 10, x /= 10;
    while (top) putchar(stackk[top--] + '0');
}
int n,m,a[maxn],d[maxn],dd[maxn],cnt;
vector<int>e[maxn];
int slove(int st) {
    int ans=0;
    rep(i,1,n) d[i]=dd[i];

    queue<int>q1,q2;
    for(int i=1 ; i<=n ; i++) {
        if(d[i]==0) {
            if(a[i]==1) q1.push(i);
            else q2.push(i);
        }
    }

    while(1) {

        if(st==1&&q1.size()==0&&q2.size()) st=2,ans++;
        else  if(st==1&&q1.size()==0&&q2.size()==0) break;
        if(st==2&&q2.size()==0&&q1.size()) ans++,st=1;
        else if(st==2&&q2.size()==0&&q1.size()==0) break;
        if(st==1) {
            int u=q1.front();
            q1.pop();
            for(int v:e[u]) {
                d[v]--;
                if(d[v]==0) {
                    if(a[v]==1) q1.push(v);
                    else q2.push(v);
                }
            }
        } else {
            int u=q2.front();
            q2.pop();
            for(int v:e[u]) {
                d[v]--;
                if(d[v]==0) {
                    if(a[v]==1) q1.push(v);
                    else q2.push(v);
                }
            }
        }

    }
    return ans;
}

int main() {
    cin>>n>>m;
    rep(i,1,n ) a[i]=read();
    for(int i=1 ; i<=m ; i++) {
        int u=read();
        int v=read();
        dd[v]++;
        e[u].push_back(v);
    }
    int ans1=slove(1);
    int ans2=slove(2);
    out(min(ans1,ans2));
    return 0;
}
/**
5 6
1 2 1 2 1
1 2
1 3
2 4
3 4
2 5
3 5
**/