[CF960G]Bandit Blues(第一类斯特林数+分治卷积)

a.png

Solution:

​ 先考虑前缀,设 \(f(i, j)\) 为长度为 \(i\) 的排列中满足前缀最大值为自己的数有 \(j\) 个的排列数。

假设新加一个数 \(i+1\) 那么会有:
\[ f(i,j)\rightarrow f(i + 1, j + 1)\\ f(i, j)\times i\rightarrow f(i + 1, j) \]
​ 即将 \(i+1\) 放在那哪个位置,会对后面产生贡献,综合一下,\(f(i, j)\) 就是第一类斯特林数 \(i \brack j\)

​ 然后再考虑后缀,不难发现,对于长度为 \(n\) 的排列,前缀为自己的一定是在 \(n\) 以及 \(n\) 的左边,后缀为自己的一定在 \(n\)\(n\) 的右边,于是可以枚举 \(n\) 的位置 \(i\),生成一个合法的方案为:先从 \(n-1\) 个数中选 \(i-1\) 个数,然后放在 \(n\) 两边,再将他们(两边互不干扰)分别分成 \(a-1, b-1\) 个环。
\[ ans=\sum_{i=1}^n{~n - 1~\choose i - 1} {i - 1\brack a - 1}{n - i\brack b - 1} \]
​ 考虑组合意义,分成两个部分,环是可以拼在一起的,于是可以改变操作的顺序,即先分环,再分边。
\[ ans = {n - 1\brack a + b - 2}{a + b - 2\choose a - 1} \]
​ 第一类斯特林数 \(n\brack i\) 的生成函数为:
\[ F_n(x) =\prod_{i\geq0}^{n-1}(x + i) \]

​ 用分治卷积快速求出一行第一类斯特林数即可。

Code

#include <vector>
#include <cmath>
#include <cstdio>
#include <cassert>
#include <cstring>
#include <iostream>
#include <algorithm>

typedef long long LL;
typedef unsigned long long uLL;

#define fir first
#define sec second
#define SZ(x) (int)x.size()
#define MP(x, y) std::make_pair(x, y)
#define PB(x) push_back(x)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define GO debug("GO\n")
#define rep(i, a, b) for (register int i = (a), i##end = (b); (i) <= i##end; ++ (i))
#define drep(i, a, b) for (register int i = (a), i##end = (b); (i) >= i##end; -- (i))
#define REP(i, a, b) for (register int i = (a), i##end = (b); (i) < i##end; ++ (i))

inline int read() {
    register int x = 0; register int f = 1; register char c;
    while (!isdigit(c = getchar())) if (c == '-') f = -1;
    while (x = (x << 1) + (x << 3) + (c xor 48), isdigit(c = getchar()));
    return x * f;
}
template<class T> inline void write(T x) {
    static char stk[30]; static int top = 0;
    if (x < 0) { x = -x, putchar('-'); }
    while (stk[++top] = x % 10 xor 48, x /= 10, x);
    while (putchar(stk[top--]), top);
}
template<typename T> inline bool chkmin(T &a, T b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }

using namespace std;

const int MOD = 998244353;
const int maxn = 1e5 + 2;

LL qpow(LL a, LL b) 
{
    LL ans = 1;
    while (b) 
    {
        if (b & 1)
            ans = ans * a % MOD;
        a = a * a % MOD;
        b >>= 1;
    }
    return ans;
}

int Inv(LL x) 
{
    return qpow(x, MOD - 2);
}

namespace Poly 
{
    const int G = 3;

    int rev[maxn * 2], omega[maxn * 2], invomega[maxn * 2];

    void init(int lim, int lg2) 
    {
        REP (i, 0, lim) rev[i] = rev[i >> 1] >> 1 | (i & 1) << (lg2 - 1);
        omega[0] = invomega[0] = 1;
        omega[1] = qpow(G, (MOD - 1) / lim);
        invomega[1] = Inv(omega[1]);
        REP (i, 2, lim) 
        {
            omega[i] = 1ll * omega[i - 1] * omega[1] % MOD;
            invomega[i] = 1ll * invomega[i - 1] * invomega[1] % MOD;
        }
    }

    void NTT(int a[], int lim, int omega[]) 
    {
        REP (i, 0, lim) if (rev[i] > i) swap(a[i], a[rev[i]]);
        for (register int len = 2; len <= lim; len <<= 1)
        {
            register int m = len >> 1;
            for (register int *p = a; p != a + lim; p += len)
                for (register int i = 0; i < m; ++i)
                {
                    register int t = 1ll * omega[lim / len * i] * p[i + m] % MOD;
                    p[i + m] = (1ll * p[i] - t + MOD) % MOD;
                    p[i] = (1ll * p[i] + t) % MOD;
                }
        }
    }
    
    void DFT(int a[], int lim)
    { NTT(a, lim, omega); }

    void IDFT(int a[], int lim)
    {
        NTT(a, lim, invomega);
        int inv = Inv(lim);
        REP (i, 0, lim) a[i] = 1ll * a[i] * inv % MOD;
    }

    void Mul(const vector<int> a, const vector<int> b, vector<int> &c)
    {
        static int A[maxn * 2], B[2 * maxn];
        int n = a.size(), m = b.size();
        int lg2 = log2(n + m) + 1;
        int lim = 1 << lg2;
        copy(a.begin(), a.end(), A);
        fill(A + n, A + lim, 0);
        copy(b.begin(), b.end(), B);
        fill(B + m, B + lim, 0);
        init(lim, lg2);
        DFT(A, lim);
        DFT(B, lim);
        REP (i, 0, lim) A[i] = 1ll * A[i] * B[i] % MOD;
        IDFT(A, lim);
        c.resize(n + m - 1);
        copy(A, A + n + m - 1, c.begin());
    }
}

vector<int> s[maxn * 4];

void solve(int o, int l, int r) 
{
    if (l == r) 
    {
        s[o].push_back(l);
        s[o].push_back(1);
        return;
    }
    int mid = (l + r) >> 1;
    solve(o << 1, l, mid);
    solve(o << 1 | 1, mid + 1, r);
    Poly::Mul(s[o << 1], s[o << 1 | 1], s[o]);
} 

int Stirling1(int n, int m) 
{
    if (m == 0) return n == 0;
    if (m < 0 || m > n) return 0;
    if (n < 0) return 0;
    solve(1, 0, n - 1);
    return s[1][m];
}

int n, a, b;

void Input()
{
    n = read(), a = read(), b = read();
}

int fac[maxn * 2];

void Init(int N)
{
    fac[0] = 1;
    rep (i, 1, N) fac[i] = 1ll * fac[i - 1] * i % MOD;
}

int combine(int n, int m)
{
    if (n < 0 || m < 0 || n < m) return 0;
    return 1ll * fac[n] * Inv(fac[m]) % MOD * Inv(fac[n - m]) % MOD;
}

void Solve()
{
    cout << 1ll * Stirling1(n - 1, a + b - 2) * combine(a + b - 2, a - 1) % MOD << endl;
}

int main() 
{
#ifndef ONLINE_JUDGE
    freopen("a.in", "r", stdin);
    freopen("a.out", "w", stdout);
#endif

    Input();

    Init(n * 2);

    Solve();

    return 0;
}

转载于:https://www.cnblogs.com/cnyali-Tea/p/11439935.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
完整版:https://download.csdn.net/download/qq_27595745/89522468 【课程大纲】 1-1 什么是java 1-2 认识java语言 1-3 java平台的体系结构 1-4 java SE环境安装和配置 2-1 java程序简介 2-2 计算机中的程序 2-3 java程序 2-4 java类库组织结构和文档 2-5 java虚拟机简介 2-6 java的垃圾回收器 2-7 java上机练习 3-1 java语言基础入门 3-2 数据的分类 3-3 标识符、关键字和常量 3-4 运算符 3-5 表达式 3-6 顺序结构和选择结构 3-7 循环语句 3-8 跳转语句 3-9 MyEclipse工具介绍 3-10 java基础知识章节练习 4-1 一维数组 4-2 数组应用 4-3 多维数组 4-4 排序算法 4-5 增强for循环 4-6 数组和排序算法章节练习 5-0 抽象和封装 5-1 面向过程的设计思想 5-2 面向对象的设计思想 5-3 抽象 5-4 封装 5-5 属性 5-6 方法的定义 5-7 this关键字 5-8 javaBean 5-9 包 package 5-10 抽象和封装章节练习 6-0 继承和多态 6-1 继承 6-2 object类 6-3 多态 6-4 访问修饰符 6-5 static修饰符 6-6 final修饰符 6-7 abstract修饰符 6-8 接口 6-9 继承和多态 章节练习 7-1 面向对象的分析与设计简介 7-2 对象模型建立 7-3 类之间的关系 7-4 软件的可维护与复用设计原则 7-5 面向对象的设计与分析 章节练习 8-1 内部类与包装器 8-2 对象包装器 8-3 装箱和拆箱 8-4 练习题 9-1 常用类介绍 9-2 StringBuffer和String Builder类 9-3 Rintime类的使用 9-4 日期类简介 9-5 java程序国际化的实现 9-6 Random类和Math类 9-7 枚举 9-8 练习题 10-1 java异常处理 10-2 认识异常 10-3 使用try和catch捕获异常 10-4 使用throw和throws引发异常 10-5 finally关键字 10-6 getMessage和printStackTrace方法 10-7 异常分类 10-8 自定义异常类 10-9 练习题 11-1 Java集合框架和泛型机制 11-2 Collection接口 11-3 Set接口实现类 11-4 List接口实现类 11-5 Map接口 11-6 Collections类 11-7 泛型概述 11-8 练习题 12-1 多线程 12-2 线程的生命周期 12-3 线程的调度和优先级 12-4 线程的同步 12-5 集合类的同步问题 12-6 用Timer类调度任务 12-7 练习题 13-1 Java IO 13-2 Java IO原理 13-3 流类的结构 13-4 文件流 13-5 缓冲流 13-6 转换流 13-7 数据流 13-8 打印流 13-9 对象流 13-10 随机存取文件流 13-11 zip文件流 13-12 练习题 14-1 图形用户界面设计 14-2 事件处理机制 14-3 AWT常用组件 14-4 swing简介 14-5 可视化开发swing组件 14-6 声音的播放和处理 14-7 2D图形的绘制 14-8 练习题 15-1 反射 15-2 使用Java反射机制 15-3 反射与动态代理 15-4 练习题 16-1 Java标注 16-2 JDK内置的基本标注类型 16-3 自定义标注类型 16-4 对标注进行标注 16-5 利用反射获取标注信息 16-6 练习题 17-1 顶目实战1-单机版五子棋游戏 17-2 总体设计 17-3 代码实现 17-4 程序的运行与发布 17-5 手动生成可执行JAR文件 17-6 练习题 18-1 Java数据库编程 18-2 JDBC类和接口 18-3 JDBC操作SQL 18-4 JDBC基本示例 18-5 JDBC应用示例 18-6 练习题 19-1 。。。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值