UVa 12887 The Soldier‘s Dilemma

题目描述

有 $N$ 名士兵，身高各不相同，他们排成一排。
指挥官会任意选出三名士兵出列，其他士兵离开，只留下这三名士兵。
如果从左到右这三名士兵的身高顺序是 短、高、中（即最矮、最高、中等），那么指挥官会生气并惩罚他们。
其他任何顺序（如高、中、短；中、短、高等）都不会被惩罚。
士兵们不想被惩罚。
问：有多少种排列方式，使得无论选择哪三名士兵，他们从左到右的顺序 永远不会 出现“短、高、中”？

输入格式

第一行一个整数 $T$（$1\le T\le 500$），表示测试用例数量。
接下来 $T$ 行，每行一个整数 $N$（$1\le N\le 5000$）。

输出格式

对于每个测试用例，输出满足条件的排列数，结果对 $10^9+7$ 取模。

样例输入

2
2
3

样例输出

2
5

问题分析

1. 问题重述

我们有 $N$ 个不同的身高，不妨用 $1,2,\dots ,N$ 表示，数值越小越矮。
一个排列 $\pi$ 表示士兵从左到右的排列顺序。

任意选出三个位置 $i<j<k$，设这三个人的身高为 $x={\pi }_i$，$y={\pi }_j$，$z={\pi }_k$。
将这三个身高按数值从小到大排序，得到 $a<b<c$，分别对应“短”、“中”、“高”。
禁止的情况是：在原排列中，这三个人的出现顺序是 短、高、中，即：

• 最矮的人 $a$ 在位置 $i$
• 最高的人 $c$ 在位置 $j$
• 中等的人 $b$ 在位置 $k$

因为 $a<b<c$，所以 $p_i<p_k<p_j$。

2. 转换为模式避免问题

在排列模式避免（$\text{Pattern Avoidance}$）理论中，三个位置 $i<j<k$ 满足 $p_i<p_k<p_j$ 被称为 $1-3-2$ 模式（也称为 $132$ 模式）。

因此，本题等价于：求长度为 $N$ 且不包含 $132$ 模式的排列个数。

这是一个经典结论：避免 $132$ 模式的排列数等于第 $N$ 个卡特兰数（$\text{Catalan Number}$）。

解题思路

3. 卡特兰数简介

卡特兰数 $C_n$ 的定义为：

$$C_0=1,C_{n+1}=\sum _{i=0}^n{C}_i\cdot C_{n-i}$$

通项公式为：

$$C_n=\frac{1}{n+1}\left(\genfrac{}{}{0ex}{}{2n}{n}\right)$$

卡特兰数在很多组合问题中出现，例如：

• 合法括号序列数
• 二叉树形态数
• 凸多边形三角剖分数
• 不越过对角线的格路路径数
• 避免 $132$ 模式的排列数

4. 为什么答案是卡特兰数？

这里给出一种直观理解（不严格证明，但有助于记忆）：

考虑排列中最大值 $N$ 的位置。如果最大值在位置 $k$，那么它左边有 $k-1$ 个数，右边有 $N-k$ 个数。
因为最大值不能出现在 132 模式中作为“高”的角色，实际上通过递推分析可以得到：

设 $f(n)$ 为长度为 $n$ 的合法排列数。考虑最大值的位置，其左边和右边的排列必须各自避免 $132$ 模式，且右边所有数必须大于左边所有数（否则会出现 $132$ 模式）。
这恰好满足卡特兰数的递推关系：

$$f(n)=\sum _{i=0}^{n-1}f(i)\cdot f(n-1-i)$$

与卡特兰数递推一致，且 $f(0)=1$，因此 $f(n)=C_n$。

5. 模运算与预处理

由于 $N\le 5000$，我们需要计算 $C_N=\frac{1}{N+1}\left(\genfrac{}{}{0ex}{}{2N}{N}\right)\mathrm{mod}M$，其中 $M=10^9+7$。

• 预处理阶乘 fact[i] = i! mod M，最大到 $2N$。
• 预处理逆元 invFact[i]，使用费马小定理 $a^{-1}\equiv a^{M-2}(\mathrm{mod}M)$。
• 组合数 $\left(\genfrac{}{}{0ex}{}{2N}{N}\right)=fact[2N]\cdot invFact[N]\cdot invFact[N]\mathrm{mod}M$。
• 最后乘以 $N+1$ 的逆元。

时间复杂度 $O(N_{\max }+T)$，空间复杂度 $O(N_{\max })$。

代码实现

// The Soldier's Dilemma
// UVa ID: 12887
// Verdict: Accepted
// Submission Date: 2026-05-27
// UVa Run Time: 0.000s
//
// 版权所有（C）2026，邱秋。metaphysis # yeah dot net

#include <bits/stdc++.h>
using namespace std;

const int MOD = 1000000007;
const int MAXN = 5000;

long long fact[MAXN * 2 + 5], invFact[MAXN * 2 + 5];

long long modPow(long long a, long long e) {
    long long res = 1;
    while (e) {
        if (e & 1) res = res * a % MOD;
        a = a * a % MOD;
        e >>= 1;
    }
    return res;
}

void precompute() {
    fact[0] = 1;
    for (int i = 1; i <= 2 * MAXN; ++i)
        fact[i] = fact[i - 1] * i % MOD;
    invFact[2 * MAXN] = modPow(fact[2 * MAXN], MOD - 2);
    for (int i = 2 * MAXN - 1; i >= 0; --i)
        invFact[i] = invFact[i + 1] * (i + 1) % MOD;
}

long long comb(int n, int k) {
    if (k < 0 || k > n) return 0;
    return fact[n] * invFact[k] % MOD * invFact[n - k] % MOD;
}

long long catalan(int n) {
    return comb(2 * n, n) * modPow(n + 1, MOD - 2) % MOD;
}

int main() {
    precompute();
    int T;
    scanf("%d", &T);
    while (T--) {
        int N;
        scanf("%d", &N);
        printf("%lld\n", catalan(N));
    }
    return 0;
}

总结

本题的关键在于将“短、高、中”这种身高顺序转化为排列中的 $132$ 模式，然后利用组合数学中的经典结论：避免 $132$ 模式的排列数等于卡特兰数。
通过预处理阶乘和逆元，可以在 $O(1)$ 时间内回答每个测试用例，总时间复杂度 $O(N_{\max }+T)$，空间复杂度 $O(N_{\max })$，完全满足题目限制。