九小时九个人九扇门 题解

题目描述

在打越钢太郎的著名解谜游戏系列《极限脱出》的第一作《九小时九个人九扇门》中，有这样一个有趣的设定：游戏中，9位主人公被困在一座大型的豪华巨轮中，每个人手上都有一个奇怪的手表，手表上有一个数字，9个人的数字分别是1−9；在巨轮中，还有很多紧闭的数字门，每扇数字门上也有一个1−9的数字，要想打开数字门逃出生天，主角们必须要满足一个奇怪的条件：

k个人能够打开门上数字为ddd的一扇数字门，当且仅当这kkk个人的腕表数字之和的数字根恰好为d。

一个数字的数字根是指：将该数字各数位上的数字相加得到一个新的数，直到得到的数字小于10为止，例如，149的数字根为149=>1+4+9=14=>1+4=5，故149的数字根为5。我们约定，小于10的数字，其数字根就为其本身。

例如，如果游戏中的一宫（手表数字为1）、四叶（手表数字为4）、八代（手表数字为8）三人组合在一起，就可以打开编号为4的数字门，这是因为1+4+8=13，而13的数字根为4。

现在，你是游戏的主角，淳平，你知道船上包括自己在内的nnn个人的手表数字，为了分析局势，你想要计算出可以打开1−9号门的人物组合有多少种，你可以完成这项任务吗？

输入描述:

输入的第一行包含一个整数n(1≤n≤105)，主人公的数量。

下面一行n个数，第i个数字ai​(1≤ai​≤109)表示第i位主人公的腕表数字。

输出描述:

你需要输出9个数字，第iii个数字表示有多少种不同的人物组合，可以打开编号为i的数字门。

答案可能很大，请你将答案对998244353取模后输出。

示例1

输入

9
1 2 3 4 5 6 7 8 9

输出

56 56 58 56 56 58 56 56 59

思路：

1、dp问题，属于01背包的变形；

2、f[i][j]表示第i个人组合成j的大小；

3、数字根：各数位的和的k进制下的数字根，相当于对（k-1）取模，但数==0||数==k-1，结果都是0；

4、gen（a+b）=gen（gen（a）+gen（b））；

代码如下：

#include<bits/stdc++.h>
using namespace std;
template<class t>inline void read(t &res){
    char c;t flag=1;
    while((c=getchar())>'9'||c<'0')if(c='-')flag=-1; res=c-'0';
    while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
    
}
const int maxj=1e5+100,M=998244353;
int a[maxj],f[maxj][10];
int main(){
   int n;
    read(n);
    for(int i=1;i<=n;++i)
        read(a[i]);
    for(int i=1;i<=n;++i){
        f[i][a[i]%9]=(f[i][a[i]%9]+1)%M;
        for(int j=0;j<=8;++j){
            f[i][j]=(f[i][j]+f[i-1][j])%M;//没有a[i]
            f[i][(j+a[i])%9]=(f[i][(j+a[i])%9]+f[i-1][j])%M;//有a[i]加入;
        }
    }
    for(int i=1;i<=8;++i)cout<<f[n][i]<<' ';
    cout<<f[n][0]<<'\n';
    return 0;
}