原题链接:https://codeforces.com/contest/1549/problem/E
题目大意
给定 $n$,$q$ 次询问。
每次询问给定一个 $x$,求 $\sum_{i=0}^n \binom{3i}{x}$
思路
考虑定义 $f_{i,j}$ 表示 $\sum_{k=0}^{n-1} \binom{3k + j}{i}$,令 $0 \leq j < m$
注意到
- $\sum_{j=0}^{m-1} f_{i,j} = \sum_{j=0}^{3n – 1} \binom{j}{i} = \binom{3n}{i + 1}$
- $f_{i,j} = f_{ i – 1, j – 1 } + f_{i, j – 1 }$
发现可以解出 $f_{i,0}$
$O(n)$ 递推即可。
Continue reading “Codeforces 1549E / 1548C”