【JSOI 2013】密码

题目大意

对于一个$m$位的十进制整数$x=(a_0a_1a_2...a_{m-1})_{10}$，定义

$g(x)=\sum_{i=0}^{m-1}a_i$

定义集合

$S_n = \{ x|0<g(x)\leq n \}$

给出$n$，求下面这个式子

$\sum_{x\in S_n} \sum_{y\in Sn, x<y}xy$

$n\leq 10^{18}$

分析

一开始看这题的时候不小心把$g(x)$里的$\sum$看成了$\prod$。囧

剩下的就是数位DP了。
记$cnt_x$表示数位和为$x$的数的个数，$linear_x$表示数位和为$x$的所有数的总和，$square_x$表示数位和为$x$的数的平方和。

通过枚举当前位选什么，不难写出递推式

$cnt_i=\sum_{j=1}^9cnt_{i-j}$
$linear_i=\sum_{j=1}^910\cdot linear_{i-j}+j\cdot cnt_{i-j}$
$square_i=\sum_{j=1}^9100\cdot square_{i-j}+20j\cdot linear_{i-j}+j^2cnt_{i-j}$

发现三个状态的转移都只与最近$9$位有关，将它们压成矩阵，用矩阵快速幂转移即可。

时间复杂度$O(log_n)$，其中常数为$29^3$