先附一组sd图

然后放上原题链接

注意,队伍不同指的是喜好不同,不是人不同

先想到$DP$,然后你会发现并没有什么优秀的状态设计,然后我们考虑容斥

设$lim$表示选的癌坤组数的上限,$f_i$为先选出来$i$组剩下随便排的方案数,那么答案就是

\(\sum\limits_{i=0}^{lim}(-1)^i\times\ f_i\)

于是问题转化为了求$f_i$。显然$f_i$可以表示为一个组合数再乘一个东西,具体来说组合数代表在$n$个同学中选$i$组癌坤的方案数,后面的那一部分代表满足限制的情况下把$n-4i$个人随便排的方案数

先考虑把组合数给搞出来,不难发现$i$组癌坤把$n$个同学分成了$i+1$段,这$i+1$段的长度之和为$n-4i$,由此我们可以得到以下不定方程

\(x_1+x_2+...+x_{i+1}=n-4i\)

那个组合数就是它的非负整数解个数,这是个经典问题,答案为$\binom{(n-4i)+(i+1)-1}{(i+1)-1}=\binom$

而后面那一部分的限制条件看起来用生成函数很好搞,同时还是个排列,考虑把$4$个$EGF$卷起来,最后后面那一部分的答案即为

\([x^{n-4i}]\sum\limits_{j=0}^{a-i}\frac{x^j}{j!}\sum\limits_{j=0}^{b-i}\frac{x^j}{j!}\sum\limits_{j=0}^{c-i}\frac{x^j}{j!}\sum\limits_{j=0}^{d-i}\frac{x^j}{j!}\)

卷积一下就出来了

至此,我们已经得到了一个$O(n^2logn)$的大常数做法了,但是好像可以在$O(n)$时间内把后面那一部分求出来(可以参考其他题解)?

放上我的超丑代码:

#include <bits/stdc++.h>

using namespace std;

#define N 1000
#define MOD 998244353 int n, cnta, cntb, cntc, cntd, lim, C[N + 5][N + 5], fac[N + 5], facinv[N + 5];
int ans, a[8 * N + 5], b[8 * N + 5], c[8 * N + 5], d[8 * N + 5]; void add(int &x, int y) {
x = (x + y) % MOD;
if (x < 0) x + MOD;
} int Add(int x, int y) {
return ((x + y) % MOD + MOD) % MOD;
} int Mul(int x, int y) {
return (1LL * x * y % MOD + MOD) % MOD;
} int fpow(int x, int p) {
int ret = 1;
while (p) {
if (p & 1) ret = Mul(ret, x);
x = Mul(x, x);
p >>= 1;
}
return ret;
} void bitReverse(int *s, int len, int bit) {
int swp[8 * N + 5];
for (int i = 1; i < len; ++i) {
swp[i] = (swp[i >> 1] >> 1) | ((i & 1) << (bit - 1));
if (i < swp[i]) swap(s[i], s[swp[i]]);
}
} void NTT(int *s, int len, int bit, int flag) {
bitReverse(s, len, bit);
for (int l = 2; l <= len; l <<= 1) {
int mid = l >> 1, t = fpow(3, (MOD - 1) / l);
if (flag == -1) t = fpow(t, MOD - 2);
for (int *p = s; p != s + len; p += l) {
int w = 1;
for (int i = 0; i < mid; ++i) {
int x = 1LL * w * p[i + mid] % MOD;
p[i + mid] = (p[i] - x) % MOD;
p[i] = (p[i] + x) % MOD;
w = 1LL * w * t % MOD;
}
}
}
if (flag == -1) {
int invlen = fpow(len, MOD - 2);
for (int i = 0; i < len; ++i)
s[i] = 1LL * s[i] * invlen % MOD;
}
} int main() {
scanf("%d%d%d%d%d", &n, &cnta, &cntb, &cntc, &cntd);
lim = min(n / 4, min(min(cnta, cntb), min(cntc, cntd)));
for (int i = 0; i <= n; ++i) C[i][0] = 1;
fac[0] = facinv[0] = 1;
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= i; ++j)
C[i][j] = Add(C[i - 1][j], C[i - 1][j - 1]);
fac[i] = Mul(fac[i - 1], i);
facinv[i] = fpow(fac[i], MOD - 2);
}
for (int i = 0, l1, l2, l3, l4, len, bit; i <= lim; ++i) {
int conv;
for (int j = 0; j <= 8 * N; ++j) a[j] = b[j] = c[j] = d[j] = 0;
for (int j = 0; j <= cnta - i; ++j) a[j] = facinv[j];
for (int j = 0; j <= cntb - i; ++j) b[j] = facinv[j];
for (int j = 0; j <= cntc - i; ++j) c[j] = facinv[j];
for (int j = 0; j <= cntd - i; ++j) d[j] = facinv[j];
l1 = cnta - i + 1, l2 = cntb - i + 1, l3 = cntc - i + 1, l4 = cntd - i + 1;
bit = 0;
while ((1 << bit) < l1 + l2 - 1) bit++;
len = 1 << bit;
NTT(a, len, bit, 1), NTT(b, len, bit, 1);
for (int j = 0; j < len; ++j) a[j] = Mul(a[j], b[j]);
NTT(a, len, bit, -1);
l1 = l1 + l2 - 1;
bit = 0;
while ((1 << bit) < l3 + l4 - 1) bit++;
len = 1 << bit;
NTT(c, len, bit, 1), NTT(d, len, bit, 1);
for (int j = 0; j < len; ++j) c[j] = Mul(c[j], d[j]);
NTT(c, len, bit, -1);
l2 = l3 + l4 - 1;
bit = 0;
while ((1 << bit) < l1 + l2 - 1) bit++;
len = 1 << bit;
NTT(a, len, bit, 1), NTT(c, len, bit, 1);
for (int j = 0; j < len; ++j) a[j] = Mul(a[j], c[j]);
NTT(a, len, bit, -1);
conv = a[n - 4 * i];
if (i & 1) add(ans, -Mul(C[n - 3 * i][i], Mul(fac[n - 4 * i], conv)));
else add(ans, Mul(C[n - 3 * i][i], Mul(fac[n - 4 * i], conv)));
}
while (ans < 0) ans += MOD;
printf("%d\n", ans);
return 0;
}

[TJOI2019]唱、跳、rap和篮球——容斥原理+生成函数的更多相关文章

  1. [bzoj5510]唱跳rap和篮球

    显然答案可以理解为有(不是仅有)0对情况-1对情况+2对情况-- 考虑这个怎么计算,先计算这t对情况的位置,有c(n-3t,t)种情况(可以理解为将这4个点缩为1个,然后再从中选t个位置),然后相当于 ...

  2. [TJOI2019]唱、跳、rap和篮球_生成函数_容斥原理_ntt

    [TJOI2019]唱.跳.rap和篮球 这么多人过没人写题解啊 那我就随便说说了嗷 这题第一步挺套路的,就是题目要求不能存在balabala的时候考虑正难则反,要求必须存在的方案数然后用总数减,往往 ...

  3. [luogu5339] [TJOI2019]唱、跳、rap和篮球(容斥原理+组合数学)(不用NTT)

    [luogu5339] [TJOI2019]唱.跳.rap和篮球(容斥原理+组合数学)(不用NTT) 题面 略 分析 首先考虑容斥,求出有i堆人讨论的方案. 可以用捆绑法,把每堆4个人捆绑成一组,其他 ...

  4. [TJOI2019]唱、跳、rap和篮球——NTT+生成函数+容斥

    题目链接: [TJOI2019]唱.跳.rap和篮球 直接求不好求,我们考虑容斥,求出至少有$i$个聚集区间的方案数$ans_{i}$,那么最终答案就是$\sum\limits_{i=0}^{n}(- ...

  5. Luogu5339 [TJOI2019]唱、跳、rap和篮球 【生成函数,NTT】

    当时看到这道题的时候我的脑子可能是这样的: My left brain has nothing right, and my right brain has nothing left. 总之,看到&qu ...

  6. [TJOI2019]唱,跳,rap,篮球(生成函数,组合数学,NTT)

    算是补了个万年大坑了吧. 根据 wwj 的题解(最准确),设一个方案 \(S\)(不一定合法)的鸡你太美组数为 \(w(S)\). 答案就是 \(\sum\limits_{S}[w(S)=0]\). ...

  7. 将Android手机无线连接到Ubuntu实现唱跳Rap

    您想要将Android设备连接到Ubuntu以传输文件.查看Android通知.以及从Ubuntu桌面发送短信 – 你会怎么做?将文件从手机传输到PC时不要打电话给自己:使用GSConnect就可以. ...

  8. 「TJOI2019」唱、跳、rap 和篮球 题解

    题意就不用讲了吧-- 鸡你太美!!! 题意: 有 \(4\) 种喜好不同的人,分别最爱唱.跳. \(rap\).篮球,他们个数分别为 \(A,B,C,D\) ,现从他们中挑选出 \(n\) 个人并进行 ...

  9. [LOJ3106][TJOI2019]唱、跳、rap和篮球:DP+生成函数+NTT+容斥原理

    分析 令\(f(i)\)表示共\(i\)组同学讨论cxk的位置的方案数(不考虑其他位置上的人的爱好),这个数组可以很容易地通过依次考虑每个位置是否是四个人中最后一个人的位置来递推求解,时间复杂度\(O ...

随机推荐

  1. 轻松搞定Vue 使用SignalR与Asp.net Core通讯

    前言 针对于Web与其他应用的的通讯,在.Net中,SignalR是一个不错的选择,在前后端没有分离的时候,直接引用对应的signalr.js文件即可: 这里主要记录Vue与Asp.netcore 前 ...

  2. 请给出一个Scala RDD的HelloWorld例子

    [学习笔记]package comimport org.apache.spark.rdd.RDDimport org.apache.spark.SparkConfimport org.apache.s ...

  3. LeetCode 答案(python)1-17

    1.给定一个整数数组和一个目标值,找出数组中和为目标值的两个数. 你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用. 示例: 给定 nums = [2, 7, 11, 15], targe ...

  4. 8.perf top系统性能分析工具

    perf 是一个调查 Linux 中各种性能问题的有力工具. # perf --help  usage: perf [--version] [--help] COMMAND [ARGS]  The m ...

  5. 【Trie】L 语言

    [题目链接]: https://loj.ac/problem/10053 [题意]: 给出n个模式串.请问文本串是由多少个模式串组成的. [题解]: 当我学完AC自动机后,发现这个题目也太简单了吧. ...

  6. 【hash表】收集雪花

    [哈希和哈希表]收集雪花 题目描述 不同的雪花往往有不同的形状.在北方的同学想将雪花收集起来,作为礼物送给在南方的同学们.一共有n个时刻,给出每个时刻下落雪花的形状,用不同的整数表示不同的形状.在收集 ...

  7. nginx浏览器开启密码验证

    如果我们在 nginx 下搭建了一些站点,但是由于站点内容或者流量的关系,我们并不想让所有人都能正常访问,那么我们可以设置访问认证.只有让用户输入正确的用户名和密码才能正常访问.效果如下: 在 ngi ...

  8. vue.js devtools图标不亮

    第一步:打开谷歌扩展程序 第二步:设置允许访问文件地址

  9. 抽奖JQ

    <!DOCTYPE html> <html lang="en"> <head>     <meta charset="UTF-8 ...

  10. LEANGOO成员

    转自:https://www.leangoo.com/leangoo_guide/leangoo_guide_member.html 1. 看板成员及权限 一个看板上的最大成员限制为200个. 看板的 ...