D - Digi Comp II （ 拓扑排序 ）

D - Digi Comp II （ 拓扑排序 ）

题意：给定M个开关，每个开关有初始状态(L或者R)，每个开关有两个走向，分别指向左边对应的开关和右边对应的开关。 一个球走到当前开关，会走向当前状态指向的方向，并且使当前的状态改变。 问N个球从1号出发，最终每个开关的状态。 给定的关系是个DAG，除了0号都有两个出度，可以看成左右儿子。

思路：模拟一下，不难发现，如果X个球经过i节点，那么将会有X/2+X%2次走向初始方向，X/2次走向另外一个方向。即是topo排序一下可以搞。

注：不一定只要1号点入度为0，题目只保证了除0之外，出度为2；没保证入度（这一点感觉题目的确误导了很多读题人）。 所以必须开始把入度为0的都加进queue去，不然有的点跑不到。

注意：拓扑排序一定要注意，刚开始需要把所有入度为0的点全push进去。

代码：

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int maxn = 5e5+10;
int state[maxn];
int num[maxn];
vector<int> G[maxn];
int du[maxn];
int n,m;

signed main()
{
    int l,r;
    char op;
    ios::sync_with_stdio(false);
    cin >> m >> n;
    for ( int i=1; i<=n; i++ ) {
        cin >> op >> l >> r;
        if ( op=='R' ) state[i] = 0;
        else if ( op=='L' ) state[i] = 1;
        G[i].push_back(l);
        G[i].push_back(r);
        du[l]++; du[r]++;
    }
    memset(num,0,sizeof(num));
    num[1] = m;
    queue<int> Q;
    for ( int i=1; i<=n; i++ ) if ( du[i]==0 ) Q.push(i);
    while ( !Q.empty() ) {
        int x = Q.front(); Q.pop();
        int t = num[x]/2;
        if ( num[x]%2==1 ) t ++;
        int d = num[x]/2;
        if ( state[x]==0 ) {
            num[ G[x][0] ] += d;
            num[ G[x][1] ] += t;
        }
        else if ( state[x]==1 ) {
            num[ G[x][0] ] += t;
            num[ G[x][1] ] += d;
        }
        du[ G[x][0] ] --;
        if ( G[x][0]!=0 && du[ G[x][0] ]==0 ) Q.push( G[x][0] );
        du[ G[x][1] ] --;
        if ( G[x][1]!=0 && du[ G[x][1] ]==0 ) Q.push( G[x][1] );
        state[x] = ( state[x] + num[x]%2 )%2;
    }
    for ( int i=1; i<=n; i++ ) {
        if ( state[i]==0 ) cout <<"R";
        else cout << "L";
    }
    cout << endl;

    return 0;
}