挑战程序设计竞赛 2.1章习题 AOJ AOJ 0033 Ball

地址 https://vjudge.net/problem/Aizu-0033

题目大意是如图的一个罐子，随即次序放入10个编号不同的球，请问能否有某种波动开关D的方法使得每个在B或者C的球的序号是从下往上递增的

输入:

第一行是一个数字N 表示测试数据的组数

后面N行是10个空格隔开的数字 表示球放入的次序

输出：

一组测试数据输出一行 如果能满足在B或者C的球的序号是从下往上递增的 输出YES 不能则输出NO

样例

Sample Input
2
3 1 4 2 5 6 7 8 9 10
10 9 8 7 6 5 4 3 2 1
Output for the Sample Input
YES
NO

解法

使用DFS 暴力遍历

对于每个球尝试放入B或者C中，并且满足球上的序号是由下至上递增

代码如下:

#include <iostream>
#include <memory.h>

using namespace std;


const int N = 20;

int arr[N];
int A[N]; int B[N];
int idxa;
int idxb;

int n, m;

int dfs(int i)
{
    if (i >= m) return 1;

    //当前求是否可以放入A或者B
    if ( arr[i] > A[idxa] ) {
        idxa++; A[idxa] = arr[i]; i++;
        if (1 == dfs(i))  return 1;
        i--; idxa--;
    }
    
    if (arr[i] > B[idxb]) {
        idxb++; B[idxb] = arr[i]; i++;
        if (1 == dfs(i))  return 1;
        i--; idxb--;
    }

    return 0;
}


int main()
{
    cin >> n;
    while (n--) {
        memset(arr, 0, sizeof(arr));
        memset(A, 0, sizeof(A));
        memset(B, 0, sizeof(B));
        idxa = 0; idxb = 0; m = 10;
        A[0] = -999; B[0] = -999;
        for (int i = 0; i < 10; i++) {
            cin >> arr[i];
        }

        int ret = dfs(0);

        if (ret) {
            cout << "YES" << endl;
        }
        else {
            cout << "NO" << endl;
        }

    }


    return 0;
}