算法训练 摆动序列
时间限制:1.0s 内存限制:512.0MB
问题描述
如果一个序列满足下面的性质,我们就将它称为摆动序列:
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 <stdio.h>
#include <stdbool.h>
int k;
bool has_used[22] = { 0 };
int arr[22] = { 0 };
int total = 0;
void swing(int pos)
{
if (pos == k)
total++;
else if (pos == 0)
{
for (int i = 1; i <= k; ++i)
{
arr[pos] = i;
has_used[i] = 1;
for (int j = 1; j <= k; ++j)
{
if (j != i)
{
arr[pos + 1] = j;
has_used[j] = 1;
swing(pos + 2);
has_used[j] = 0;
}
}
has_used[i] = 0;
}
}
else
{
total++;
if (arr[pos - 2] < arr[pos - 1])
{
for (int i = 1; i < arr[pos - 2]; ++i)
{
if (!has_used[i])
{
arr[pos] = i;
has_used[i] = 1;
swing(pos + 1);
has_used[i] = 0;
}
}
}
else
{
for (int i = arr[pos - 2] + 1; i <= k; ++i)
{
if (!has_used[i])
{
arr[pos] = i;
has_used[i] = 1;
swing(pos + 1);
has_used[i] = 0;
}
}
}
}
}
int main()
{
scanf("%d", &k);
swing(0);
printf("%d", total);
return 0;
}
方法二:事实上,满足题意的摆动序列可以通过如下方式构造:从1,2,...,k这k个数中任意选出s个()数,且按照要求4摆放。我们可以列举出s = 2,3,4,5时序列的形状:
不难看出,生成的“摆动序列”的形状是越往右侧摆动的幅度越大,即最后一段的两个端点一定是选取出的s个数中的最小值和最大值,然后对于剩余的s-2个数,递归的按照要求4摆放,而这种摆放方式事实上是唯一的。所以只要选定s个不同的数,其摆放顺序就已经固定了。于是,题目要求的1~k这k个数中任取s(s = 2~k)个数所组成的摆动序列的总种数为:
#include <stdio.h>
int main()
{
int k;
scanf("%d", &k);
printf("%d", ((1 << k) - k - 1) << 1);
return 0;
}