B. Curiosity Has No Limits

题目链接：http://codeforces.com/problemset/problem/1072/B

 

题意：

       给出长度为n-1的两个数组a和b，要求找出一个长度为n的数组t，使得t[i]|t[i+1]=a[i] && t[i]&t[i+1]=b[i]，问是否存在这样的数组t

　　第一行输入一个n ( 2 < = n < = 1e5) 表示t数组的长度

　　第二行输入n-1个数 a1,a2……an-1 （0 < = ai < =3)

　　第三行输入n-1个数b1,b2……bn-1 (0 < = bi < =3)

 

思路：

这居然是个 dfs 的题目。因为 a[i] 和 b[i] 的范围在[0,3] ，直接枚举 t 的值就可以过了？！

 

#include <iostream>
#include <algorithm>
#include <stdlib.h>
#include <cstring>
#include <string>
#include <string.h>
#include <set>
#include <queue>
#include <stdbool.h>

#define LL long long
using namespace std;
const int maxn = 1e5 + 10;


int a[maxn],b[maxn],t[maxn];
int n;

void DFS(int x){
    if (x<1){
        printf("YES\n");
        for (int i=1;i<n;i++){
            printf("%d ",t[i]);
        }
        printf("%d\n",t[n]);
        exit(0);
    }
    for (int i=0;i<=3;i++){
        if (((i|t[x+1]) == a[x]) && ((i&t[x+1]) == b[x])){
            t[x] = i;
            DFS(x-1);
        }
    }
}


int main(){
    scanf("%d",&n);
    for (int i=1;i<n;i++){
        scanf("%d",&a[i]);
    }
    for (int i=1;i<n;i++){
        scanf("%d",&b[i]);
    }
    for (int i=0;i<=3;i++){
        t[n] = i;
        DFS(n-1);
    }
    printf("NO\n");
    return 0;
}