2022牛客寒假算法基础集训营1 A

题目：

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

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

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

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

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

思路：

可以证明每个人的数字之和的数字根和每个人数字根之和是相等的。那么我们就只需要枚举0 到 9 的情况，用滚动数组可以过。

代码：

#include <bits/stdc++.h>
#define MOD 998244353
using namespace std;
int ans[2][10];
int addsum(int need)
{
    int ans=0;
    do{
        while(need)
        {
            ans+=need%10;
            need /= 10;
        }
        need = ans;
        ans = 0;
    }while(need>9);
    return need;
}
int main()
{
    int N;
    int i,j,k;
    cin>>N;
    for(i=1;i<=N;i++)
    {
       cin>>k;
       k = addsum( k );
       for( j = 1 ; j <= 9 ; j++ ) ans[ i % 2 ][ j ] = ans[ ( i + 1 ) % 2 ][ j ];
       for(j=1;j<=9;j++)
       {
           ans[ i % 2 ][ addsum( j + k ) ] += ans[ (i+1) % 2][ j ];
           ans[ i % 2 ][ addsum( j + k ) ] %= MOD;
       }
       ans[ i % 2 ][ k ] ++ ;
       ans[ i % 2 ][ k ] %= MOD;
    }
    for(j=1;j<=9;j++) cout << ans[ ( i + 1 ) % 2 ][ j ] << ' ';
    cout << endl ;
}