Codeforces 1031 problem B——Curiosity Has No Limits

• 题目链接 ：http://codeforces.com/contest/1031/problem/B
• 题意 ：给你两个数组a和b，他们的长度是n-1，问有没有一个ans数组满足 ans[i] | ans[i-1] = a[i-1] 同时 ans[i] & ans[i-1] = b[i-1] (i > 2)。
• 思路 ：因为题面给出了 0<=ai<=3 0<=bi<=3，所以不用推公式。。很简单的暴力就可以。因为最大是3，如果比3大，那么或运算之后肯定也比3大，所以ans就考虑0—3.
• 代码 ：
  暴力多重循环有点不会。。。就用了个dfs

#include "bits/stdc++.h"
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define fori(i,l,u) for(int i = l;i < u;i++)
#define forj(j,l,u) for(int j = l;j < u;j++)
#define F first
#define S second
#define pb push_back
#define mk make_pair
typedef long long  ll;
typedef pair<int, int> pi;
typedef pair<string, int> ps;
typedef vector<int> vi;
typedef vector<string> vs;
typedef vector<pi> vpi;
const int maxn = 1e5 + 6;
int a[maxn],b[maxn];
int n;
int ans[maxn];
bool exist;
void dfs(int num){
    if (num == n+1) {
        cout<<"YES"<<endl;
        fori(i, 1, n+1) cout<<ans[i]<<" ";
        cout<<endl;
        exist = true;
        return;
    }
    for (ans[num] = 0; ans[num]<=3; ans[num]++) {
        if (((ans[num]|ans[num-1]) == a[num-1]) && ((ans[num]&ans[num-1]) == b[num-1]) ) {
//            if (num == 2) {
//                cout<<"ans[1] = "<<ans[num-1]<<" ";
//            }
//            cout<<"ans["<<num<<"]"<<" = "<<ans[num]<<endl;
            dfs(num+1);
        }
    }
}
int main()
{
//    freopen("1.txt", "r", stdin);
    while (cin>>n) {
        mem(ans,0);
        exist = false;
        fori(i, 1, n){
            cin>>a[i];
        }
        fori(i, 1, n){
            cin>>b[i];
        }
        for (ans[1] = 0; ans[1]<=3; ans[1]++) {
            dfs(2);
            if (exist) {
                break;
            }
        }
        if (!exist) {
            cout<<"NO"<<endl;
        }
    }
    return 0;
}

• 遇到的问题 ：一开始很纳闷，为啥在dfs里面满足条件输出ans数组是正确答案，而在main里面输出会是 1 4 4 4.后来想到，num = n-1之后，return，然后会返回以前的dfs里继续执行for循环里ans[num]++直到ans[num] > 3，跳出循环。所以最后都是1 4 4 4. 如果想在main里输出正确答案，就在每次dfs之后加一个break跳出循环即可。