CF1954D Colored Balls

题目描述

有 $n$ 种不同颜色的球,第 $i$ 种颜色的球有 $a_i$ 个。 这些球可以被组合成若干组。每组最多包含 $2$ 个球,并且每组中每种颜色的球最多只能有 $1$ 个。 考虑所有 $2^n$ 种颜色集合。对于某个颜色集合,定义其值为用这些颜色的球能分成的最少组数。例如,若有三种颜色,球的数量分别为 $3$、$1$ 和 $7$,则这些球最少能组合成 $7$ 组(且不能更少),所以该颜色集合的值为 $7$。 你的任务是计算所有 $2^n$ 种颜色集合的值之和。由于答案可能很大,请输出其对 $998\,244\,353$ 取模的结果。

输入格式

第一行包含一个整数 $n$($1 \le n \le 5000$),表示颜色的数量。 第二行包含 $n$ 个整数 $a_1, a_2, \dots, a_n$($1 \le a_i \le 5000$),表示第 $i$ 种颜色的球的数量。 输入的额外限制:所有球的总数不超过 $5000$。

输出格式

输出一个整数,表示所有 $2^n$ 种颜色集合的值之和,对 $998\,244\,353$ 取模。

说明/提示

考虑第一个样例。共有 $8$ 个颜色集合: - 空集,值为 $0$; - 集合 $\{1\}$,值为 $1$; - 集合 $\{2\}$,值为 $1$; - 集合 $\{3\}$,值为 $2$; - 集合 $\{1,2\}$,值为 $1$; - 集合 $\{1,3\}$,值为 $2$; - 集合 $\{2,3\}$,值为 $2$; - 集合 $\{1,2,3\}$,值为 $2$。 因此,所有 $2^n$ 种颜色集合的值之和为 $11$。 由 ChatGPT 4.1 翻译