题目链接:点击打开链接
题目大意:有A和B两个门,每个门上有一个数字,现在有n个人,每个人手里也有一个数字,现在n个人都要选择一个门进去,要求计算门内的人的数字每一位的和,一直累加到一位。这个数和门代表的数字是相同的。问n个人有多少种走法。
例如 65536 -> 25 -> 7
输入n A B,之后一行输入n个数字代表n个人手中的数字。每个数字都是大于0的
输出种类个数。
对于累加每一位的和到一位的值 = 初始数的每一位的和%9,其中0代表累加的结果是9
首先判断所有人的和%9,是不是(A+B)%9,如果不同,那么就是说不会存在人们分开到达A和B的情况,只有可能所有人都进一个门,或者都不进。
如果想同,那么如果一些人满足进入A的条件,那么剩下的人一定都满足进入B的,所以统计能进入A中个数。
dp[i][j]当第i个人是否进入A内,j为A门内所有人的和%9的结果。最终dp[n][A]就是所有的结果。
注意:对a[i]、A、B进行对9取余。
#include <cstdio> #include <cstring> #include <algorithm> using namespace std ; #define Mod 258280327 #define LL __int64 LL dp[100010][10] ; int a[100010] ; int main() { int t , i , j , n , A , B , sum ; scanf("%d", &t) ; while( t-- ) { scanf("%d %d %d", &n, &A, &B) ; memset(dp,0,sizeof(dp)) ; A %= 9 ; B %= 9 ; for(i = 1 , sum = 0 ; i <= n ; i++) { scanf("%d", &a[i]) ; a[i] %= 9 ; sum = (sum+a[i]) % 9 ; } if( (A+B)%9 != sum ) { if( A == sum ) dp[0][0]++ ; if( B == sum ) dp[0][0]++ ; printf("%I64d\n", dp[0][0]) ; continue ; } dp[0][0] = 1 ; for(i = 1 ; i <= n ; i++) { for(j = 0 ; j < 10 ; j++) { dp[i][j] += dp[i-1][j] ; dp[i][ (j+a[i])%9 ] += dp[i-1][j] ; dp[i][ (j+a[i])%9 ] %= Mod ; } } printf("%I64d\n", dp[n][A]%Mod ) ; } return 0 ; }
版权声明:转载请注明出处:http://blog.csdn.net/winddreams
时间: 2024-11-05 13:31:02