蓝桥杯 算法训练 摆动序列(基于搜索算法)

问题描述
  如果一个序列满足下面的性质,我们就将它称为摆动序列:
  1. 序列中的所有数都是不大于 k的正整数;
  2. 序列中至少有两个数。
  3. 序列中的数两两不相等;
  4. 如果第 i – 1个数比第 i – 2个数大,则第 i个数比第 i – 2个数小;如果第 i – 1个数比第 i – 2个数小,则第 i个数比第 i – 2个数大。
  比如,当 k = 3时,有下面几个这样的序列:
  1 2
  1 3
  2 1
  2 1 3
  2 3
  2 3 1
  3 1
  3 2
  一共有8种,给定 k,请求出满足上面要求的序列的个数。
输入格式
  输入包含了一个整数 k。( k<=20)
输出格式
  输出一个整数,表示满足要求的序列个数。
样例输入
3
样例输出
8

大蓝桥官网上非说这道题是dp,emmmm。。。反正我找了半天没找到状态转移方程,然后受到八皇后的启发,其实搜索算法加回溯很容易就能解出来,虽然可能时间上稍慢,但测试用例也就到20,很符合暴力杯的特点ahhhh然后代码如下:

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int value[25],used[25];
int tot=0;
int n;
void dfs(int x)
{
    if(x>n)
    return ;
    if(x==1||x==2)   //x等于一时属于被迫计算,回来再减去,等于二时所有序列都是摆动序列,所以特判。
    {
        for(int i=1;i<=n;i++)
        {
            if(used[i]==0)
            {
                used[i]=1;
                value[x]=i;
                ++tot;
                dfs(x+1);
                used[i]=0;     //搜索完毕后把变量改回来,一定要改回来
                value[x]=0;
            }
        }
    }
    else
    {
            if(value[x-1]>value[x-2])   //按照题目要求分情况讨论继续搜索
            {
                for(int i=1;i<value[x-2];i++)
                {
                    if(used[i]==0)
                    {
                        used[i]=1;
                        value[x]=i;
                        ++tot;
                       dfs(x+1);
                       used[i]=0;
                       value[x]=0;
                    }
                }
            }
            else if(value[x-1]<value[x-2])
            {
                for(int i=value[x-2]+1;i<=n;i++)
                {
                    if(used[i]==0)
                    {
                        used[i]=1;
                        value[x]=i;
                        ++tot;
                       dfs(x+1);
                       used[i]=0;
                       value[x]=0;
                    }
                }
            }
    }
}
int main()
{
   scanf("%d",&n);
   memset(used,0,sizeof(used));
   memset(value,0,sizeof(value));
   dfs(1);
   printf("%d",tot-n);  //把x=1的情况减去

}

我们在搜索过程中记录状态的个数,即为最终所求,除了x=1时,搜索过程中每个状态都属于摆动序列哦,仔细体会下吧。

猜你喜欢

转载自blog.csdn.net/langzitan123/article/details/79904675