187. 导弹防御系统/dfs+剪枝【AcWing】

187. 导弹防御系统

为了对抗附近恶意国家的威胁，R 国更新了他们的导弹防御系统。

一套防御系统的导弹拦截高度要么一直 严格单调 上升要么一直 严格单调 下降。

例如，一套系统先后拦截了高度为 3 和高度为 4 的两发导弹，那么接下来该系统就只能拦截高度大于 4 的导弹。

给定即将袭来的一系列导弹的高度，请你求出至少需要多少套防御系统，就可以将它们全部击落。

输入格式
输入包含多组测试用例。

对于每个测试用例，第一行包含整数 n，表示来袭导弹数量。

第二行包含 n 个不同的整数，表示每个导弹的高度。

当输入测试用例 n=0 时，表示输入终止，且该用例无需处理。

输出格式
对于每个测试用例，输出一个占据一行的整数，表示所需的防御系统数量。

数据范围
1≤n≤50

输入样例：
5
3 5 2 4 1
0
输出样例：
2

样例解释
对于给出样例，最少需要两套防御系统。

一套击落高度为 3,4 的导弹，另一套击落高度为 5,2,1 的导弹。

dfs+剪枝

思路：首先每次每个子弹有两个选择，放到上升数组或者放到下降数组，如果放不了则开一个新的，代码中的操作过程就比较巧妙，然后if(b+s>=ans) return ;这个剪枝操作很重要！！！！如果已经大于等于现有值，则直接返回，且少了等于号是会超时的…，因为这题时间要求比较高，所以多了个等于就少了一步所以会节省一点时间…，这一题不用memset也是可以的，因为每次dfs重置了bigger和smaller数组的上界，因此不会越界访问到上次留下来的啦

#include<bits/stdc++.h>
using namespace std;
int a[55],bigger[55],smaller[55];
int ans=55;
void dfs(int idx,int n,int b,int s)
{
    if(b+s>=ans) return ;
    if(idx==n)
    {
        ans=b+s;
        return ;
    }
    int i;
    for(i=0;i<b;i++)
    {
        if(a[idx]>bigger[i])
        break;
    }
    int x=bigger[i];
    bigger[i]=a[idx];
    dfs(idx+1,n,max(i+1,b),s);
    bigger[i]=x;
    for(i=0;i<s;i++)
    {
        if(a[idx]<smaller[i])
        break;
    }
    x=smaller[i];
    smaller[i]=a[idx];
    dfs(idx+1,n,b,max(i+1,s));
    smaller[i]=x;
}
int main()
{
    int n;
    while(scanf("%d",&n)&&n!=0)
    {
        for(int i=0;i<n;i++) cin>>a[i];
        ans=55;
        dfs(0,n,0,0);
        printf("%d\n",ans);
    }
    return 0;
}