fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. signed main(){
  4. 	ios_base::sync_with_stdio(false);
  5.     cin.tie(NULL);
  6. 	int n;
  7. 	cin >> n;
  8. 	vector<int> vec(n);
  9. 	for(int i = 0; i < n; i++){
  10. 		cin >> vec[i];
  11. 	}
  12.  
  13. 	vector<vector<int>> dp(n + 1, vector<int>(10, 0));
  14. 	dp[0][vec[0]] = 1;
  15. 	for(int i = 0; i < n; i++){
  16. 		for(int j = 0; j < 10; j++){
  17. 			if(i < n - 1 && dp[i][j] > 0){
  18. 			   dp[i + 1][(j + vec[i + 1]) % 10] += dp[i][j] % 998244353; 
  19.    			   dp[i + 1][(j * vec[i + 1]) % 10] += dp[i][j] % 998244353; 
  20.  
  21.  
  22. 			}
  23. 		}
  24. 	}
  25. 	for(int i = 0; i < 10; i++){
  26. 		cout << dp[n - 1][i] << endl; 
  27. 	}
  28. }
Success #stdin #stdout 0.01s 5320KB
comments (?)
stdin
copy 
5
0 1 2 3 4
stdout
copy 
6
0
1
1
4
0
1
1
0
2
https://ideone.com/OaN57A
language:
C++14 (gcc 8.3)
created:
2 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!