题目描述

庭院里有一棵古树。
圣诞节到了,我想给古树做点装饰,给他一个惊喜。
他会不会喜欢呢?
这棵树可以分为$n$层,第$i$层有$a_i$个防治装饰品的位置,有$m$种颜色的装饰品可供选择。
为了能让他喜欢,我想让装饰品满足以下条件:
$1.$在同一层两个相邻的装饰品不能有同一种颜色;
$2.$相邻两层的颜色集合不能相同(这里颜色集合是指这一层所有出现的颜色去重之后的集合,也就是每个颜色只在集合被最多出现一次)。
我想知道,有多少种不同的方案能让他满意呢?由于方案书可能很多,请告诉我模$p$之后的答案。
谢谢你。


输入格式

第一行三个整数,$n,m,p$。
第二行$n$个整数,第$i$个整数表示$a_i$。


输出格式

一行一个整数,表示答案。


样例

样例输入:

3 2 100
3 1 2

样例输出:

8


数据范围与提示

样例解释:

如下集中方法满足条件:
$1.121|1|12$
$2.121|1|21$
$3.121|2|12$
$4.121|2|21$
$5.212|1|12$
$6.212|1|21$
$7.212|2|12$
$8.212|2|21$

数据范围:

记$S=\sum \limits_{i=1}^n a_i$。
对于$20\%$的数据,$1\leqslant m\leqslant 3,S\leqslant 10$。
对于$50\%$的数据,$1\leqslant 1,000,1\leqslant m\leqslant 10$。
对于$100\%$的数据,$1\leqslant n\leqslant {10}^6,1\leqslant m\leqslant {10}^6,1\leqslant a_i\leqslant 5,000,S\leqslant {10}^7,2\leqslant p\leqslant {10}^9$。


题解

考虑$DP$。

定义$dp[i][j]$表示到了第$i$层,放了$j$个颜色的装饰品的方案数。

定义$g[i][j]$表示对于一行,到了第$i$个位置,放了$j$个颜色的装饰品的方案数。

我们可以先不管都选了那种颜色,最后在乘上一个组合数就好了。
因为总颜色数是不变的,这样就方便了许多。

$g$数组的转移:

  $g[i][j]=g[i-1][j-1]*j+g[i-1][j]*(j-1)$。

$dp$数组的转移:

  $dp[i][j]=dp[i-1][0]*g[a[i]][j]*C[j]-dp[i-1][j]*g[a[i]][j]$

  $dp[i][0]=dp[i][0]+dp[i][j]$

$dp$数组第一维滚动即可。

时间复杂度:$\Theta(S)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
int n,m,p;
int a[1000001],maxn,pri[1000001],t[1000001];
bool vis[10000001],pos;
long long dp[2][5001],g[5001][5001],C[5001];
int main()
{
scanf("%d%d%d",&n,&m,&p);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
maxn=max(maxn,a[i]);
}
for(int i=2;i<=m;i++)
{
if(!vis[i])pri[++pri[0]]=i;
for(int j=1;j<=pri[0];j++)
{
if(i*pri[j]>m)break;
vis[i*pri[j]]=1;
if(!(i%pri[j]))break;
}
}
dp[0][0]=g[0][0]=1;
for(int i=1;i<=maxn;i++)
for(int j=1;j<=min(m,maxn);j++)
g[i][j]=(g[i-1][j-1]*j%p+g[i-1][j]*(j-1)%p)%p;
for(int i=1;i<=min(m,maxn);i++)
{
int flag1=i,flag2=m-i+1;
for(int j=1;j<=pri[0]&&pri[j]<=flag1;j++)
while(!(flag1%pri[j]))
{
flag1/=pri[j];
t[j]--;
}
for(int j=1;j<=pri[0]&&pri[j]<=flag2;j++)
while(!(flag2%pri[j]))
{
flag2/=pri[j];
t[j]++;
}
C[i]=1;
for(int j=1;j<=pri[0]&&pri[j]<=m;j++)
for(int k=1;k<=t[j];k++)
C[i]=(C[i]*pri[j])%p;
}
for(int i=1;i<=n;i++)
{
pos^=1;
memset(dp[pos],0,sizeof(dp[pos]));
for(int j=1;j<=min(m,a[i]);j++)
{
dp[pos][j]=(dp[pos^1][0]*g[a[i]][j]%p*C[j]%p-dp[pos^1][j]*g[a[i]][j]%p+p)%p;
dp[pos][0]=(dp[pos][0]+dp[pos][j])%p;
}
}
printf("%lld",dp[pos][0]);
return 0;
}

rp++

[CSP-S模拟测试]:长寿花(DP+组合数)的更多相关文章

  1. [CSP-S模拟测试]:matrix(DP)

    题目描述 求出满足以下条件的$n\times m$的$01$矩阵个数:(1)第$i$行第$1~l_i$列恰好有$1$个$1$.(2)第$i$行第$r_i~m$列恰好有$1$个$1$.(3)每列至多有$ ...

  2. [CSP-S模拟测试]:tree(DP)

    题目传送门(内部题57) 输入格式 第一行包含一个数:$n$表示树的节点数.接下来$n-1$行,每行包含两个数:$u,v$表示无根树的一条边. 输出格式 输出$n$行,第$i$行包含一个浮点数,保留三 ...

  3. [CSP-S模拟测试]:B(DP+数学)

    题目传送门(内部题45) 输入格式 第一行$3$个整数$n,m,P$.第二行$m$个整数,表示$m$次询问. 输出格式 一行$m$个整数表示答案. 样例 样例输入1: 2 4 40 1 2 3 样例输 ...

  4. [CSP-S模拟测试]:蛇(DP+构造+哈希)

    题目传送门(内部题140) 输入格式 前两行有两个长度相同的字符串,描述林先森花园上的字母. 第三行一个字符串$S$. 输出格式 输出一行一个整数,表示有多少种可能的蛇,对$10^9+7$取模. 样例 ...

  5. [CSP-S模拟测试]:最小值(DP+乱搞)

    题目背景 $Maxtir$更喜欢序列的最小值. 题目传送门(内部题128) 输入格式 第一行输入一个正整数$n$和四个整数$A,B,C,D$. 第二行输入$n$个整数,第$i$个数表示$a_i$. 输 ...

  6. [CSP-S模拟测试]:计数(DP+记忆化搜索)

    题目描述 既然是萌萌哒$visit\text{_}world$的比赛,那必然会有一道计数题啦!考虑一个$N$个节点的二叉树,它的节点被标上了$1\sim N$的编号.并且,编号为$i$的节点在二叉树的 ...

  7. [CSP-S模拟测试]:题(DP+数学)

    题目描述 出个题就好了.这就是出题人没有写题目背景的原因.你在平面直角坐标系上.你一开始位于$(0,0)$.每次可以在上/下/左/右四个方向中选一个走一步.即:从$(x,y)$走到$(x,y+1),( ...

  8. [CSP-S模拟测试]:题(DP)

    题目描述 由于出题人赶时间所以没办法编故事来作为背景.一开始有$n$个苹果,$m$个人依次来吃苹果,第$i$个人会尝试吃$u_i$或$v_i$号苹果,具体来说分三种情况.$\bullet 1.$两个苹 ...

  9. [CSP-S模拟测试]:y(DP+bitset)

    题目背景 $\frac{1}{4}$遇到了一道水题,叕完全不会做,于是去请教小$D$.小$D$懒得理$\frac{1}{4}$,直接就离开了.于是,$\frac{1}{4}$只好来问你,这道题是这样的 ...

随机推荐

  1. CodeIgniter 技巧 - 通过 Composer 安装 CodeIgniter 框架并安装依赖包

    PHP 项目中,通过 Composer 来管理各种依赖包,类似 Java 中的 Maven,或 Node 中的 npm.CodeIgniter 框架要想通过 Composer 自动加载包也很简单,步骤 ...

  2. Linux多线程服务器端编程

    目录 Linux多线程服务器端编程 线程安全的对象生命期管理 对象的销毁线程比较难 线程同步精要 借shared_ptr实现写时拷贝(copy-on-write) 多线程服务器的适用场合与常用编程模型 ...

  3. JSP基础--三大指令

    JSP指令 1       JSP指令概述 JSP指令的格式:<%@指令名 attr1=”” attr2=”” %>,一般都会把JSP指令放到JSP文件的最上方,但这不是必须的. JSP中 ...

  4. 第五周课程总结&实验报告三

    1.已知字符串:"this is a test of java".按要求执行以下操作:(要求源代码.结果截图.) •统计该字符串中字母s出现的次数. •统计该字符串中子串" ...

  5. (1)leetcode刷题Python笔记——两数之和

    题目如下: 给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标. 你可以假设每种输入只会对应一个答案.但是,你不能重复利用这个数 ...

  6. 基于Select模型的Windows TCP服务端和客户端程序示例

    最近跟着刘远东老师的<C++百万并发网络通信引擎架构与实现(服务端.客户端.跨平台)>,Bilibili视频地址为C++百万并发网络通信引擎架构与实现(服务端.客户端.跨平台),重新复习下 ...

  7. golang的数据类型之浮点类型

    [root@node3 shangxuetang]# cat float.go package main import "fmt" //演示golang中小数类型使用func ma ...

  8. 微信小程序(三)--小程序UI开发

    一.UI介绍 所谓的UI(user Interface)开发指的就是小程序应用界面的开发,在小程序开发框架中会为我们提供一系列的基础组件,例如HTML开发中为我们所提供的一些最基础的标签.需要注意的是 ...

  9. 线程休眠只会用 Thread.sleep?来,教你新姿势!

    线程休眠是 Java 开发经常会用到的一个手段,就是让当前线程睡一会儿,睡醒之后再继续运行. 咱大多数程序员,多线程虽然学得不好,但线程休眠,无人不知,无人不晓,也都会用,不就是用 Thread.sl ...

  10. Manacher(输出最长回文串及下标)

    http://acm.hdu.edu.cn/showproblem.php?pid=3294 Girls' research Time Limit: 3000/1000 MS (Java/Others ...