Codeforces#499 F. Mars rover (两遍dfs)

题目：http://codeforces.com/contest/1011/problem/F

题意：
给一颗逻辑树，求每个输入端改变(0<->1)后，输出端(1)的值

分析：
两遍dfs
第一遍求各点状态
第二遍求各点状态改变后是否反转output(节点1的状态)

代码：

#include <bits/stdc++.h>
using namespace std;
const int tmax=1e6+5;
int n,data[tmax][5];    //in0 and1 or2 xor3 not4
char s[10];
bool f[tmax],rev[tmax];
void dfs(int x)
{
    if(data[x][0]==0)
        f[x]=data[x][1];
    else if(data[x][0]==4)
    {
        dfs(data[x][1]);
        f[x]=!f[data[x][1]];
    }
    else
    {
        dfs(data[x][1]);
        dfs(data[x][2]);
        if(data[x][0]==1)
            f[x]=f[data[x][1]] and f[data[x][2]];
        else if(data[x][0]==2)
            f[x]=f[data[x][1]] or f[data[x][2]];
        else  f[x]=f[data[x][1]] xor f[data[x][2]];
    }
    return;
}
void test()
{
    int i;
    for(i=1;i<=n;i++)
        printf("%d:%d\n",i,f[i]);
    cout<<endl;
    for(i=1;i<=n;i++)
        printf("%d:%d\n",i,rev[i]);

    return;
}
void dfs2(int x)
{
    if(data[x][0]==0) return;
    if(data[x][0]==4)
    {
        rev[data[x][1]]=rev[x];
        dfs2(data[x][1]);
    }
    else
    {
        if(data[x][0]==1)
        {
            if(f[data[x][2]]==true) rev[data[x][1]]=rev[x];
            if(f[data[x][1]]==true) rev[data[x][2]]=rev[x];
        }
        else if(data[x][0]==2)
        {
            if(f[data[x][2]]==false) rev[data[x][1]]=rev[x];
            if(f[data[x][1]]==false) rev[data[x][2]]=rev[x];
        }
        else rev[data[x][1]]=rev[data[x][2]]=rev[x];
        dfs2(data[x][1]);
        dfs2(data[x][2]);
    }

    return;
}
int main()
{
    scanf("%d",&n);
    int i;
    for(i=1;i<=n;i++)
    {
        scanf("%s",s);
        if(s[0]=='N')
        {
            data[i][0]=4;
            scanf("%d",&data[i][1]);
        }
        else if(s[0]=='I')
            scanf("%d",&data[i][1]);
        else
        {
            scanf("%d%d",&data[i][1],&data[i][2]);
            if(s[0]=='A') data[i][0]=1;
            else if(s[0]=='O') data[i][0]=2;
            else if(s[0]=='X') data[i][0]=3;
        }
    }
    rev[1]=true;
    dfs(1);
    dfs2(1);
    for(i=1;i<=n;i++)
        if(data[i][0]==0)
    {
        if(rev[i]==true) printf("%d",!f[1]);
        else printf("%d",f[1]);
    }
    //test();
    return 0;
}