蓝桥杯 算法训练 摆动序列

题目链接：http://lx.lanqiao.cn/problem.page?gpid=T44

问题描述

如果一个序列满足下面的性质，我们就将它称为摆动序列：
　　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

解题思路：

按位置进行深度优先遍历

//个人习惯使用的头文件
#include<cstdio> 
#include<cstdlib> 
#include<cstring>
#include<iostream>  
#include<algorithm>
using namespace std;

//a[i]数组用来记录数据i这个数据是否已经被使用，c[j]数组用来存放第j个位置应该放的数
int a[100], c[100];
int n, ans = 0;

//判断是否位置pos处放数据num是否符合要求
bool judge(int num, int pos){
	if(pos == 1 || pos == 2) return 1;
	if(c[pos-1] > c[pos-2] && num < c[pos-2]) return 1;
	if(c[pos-1] < c[pos-2] && num > c[pos-2]) return 1;
	return 0;
	
}

//对每一位置进行搜索
void dfs(int pos){
	if(pos > n) return;
	for(int i = 1; i <= n; i++){
		if(!a[i] && judge(i, pos)){
			a[i] = 1;
			c[pos] = i;
			if(pos >= 2){
				ans++;
			}
			
			dfs(pos+1);
			a[i] = 0;
		}
	}
}

int main(){
	memset(a, 0, sizeof(a));
	memset(c, 0, sizeof(c));
	cin >> n;
	dfs(1);
	cout << ans << endl;
	return 0;
}