CF1845E Boxes and Balls

题目描述

有 $n$ 个盒子排成一行,编号从 $1$ 到 $n$。有些盒子里有一个球,其余盒子是空的。至少有一个盒子里有球,至少有一个盒子是空的。 每次操作,你必须选择一个有球的盒子和一个相邻的空盒子,并将球从一个盒子移动到另一个盒子。对于所有 $i$ 从 $1$ 到 $n-1$,盒子 $i$ 和 $i+1$ 被认为是相邻的。盒子 $1$ 和 $n$ 不相邻。 在恰好进行了 $k$ 次操作后,可能存在多少种不同的球的排列方式?如果存在至少一个盒子在一种排列中有球而在另一种排列中没有球,则这两种排列被认为是不同的。 由于答案可能很大,请输出答案对 $10^9+7$ 取模后的结果。

输入格式

第一行包含两个整数 $n$ 和 $k$($2 \le n \le 1500$;$1 \le k \le 1500$),分别表示盒子的数量和操作次数。 第二行包含 $n$ 个整数 $a_1, a_2, \dots, a_n$($a_i \in \{0, 1\}$),$0$ 表示该盒子为空,$1$ 表示该盒子里有一个球。至少有一个 $0$,至少有一个 $1$。

输出格式

输出一个整数,表示恰好进行了 $k$ 次操作后,可能存在的不同球的排列方式的数量,对 $10^9+7$ 取模。

说明/提示

在第一个样例中,可能的排列有: - 0 1 1 0 —— 将球从盒子 $1$ 移动到盒子 $2$ 得到; - 1 0 0 1 —— 将球从盒子 $3$ 移动到盒子 $4$ 得到; - 1 1 0 0 —— 将球从盒子 $3$ 移动到盒子 $2$ 得到。 在第二个样例中,可能的排列有: - 1 0 1 0 —— 有三种方式得到:只需将第一次操作反向执行即可; - 0 1 0 1 —— 从前两种排列中的任意一个在第一次操作后得到。 由 ChatGPT 4.1 翻译