题解 | #小红的序列乘积2.0#

小红的序列乘积2.0

https://ac.nowcoder.com/acm/contest/87656/E

小红的序列乘积2.0

思路 :

动态规划,令 表示所有考虑前 个元素的子序列种构成的题意中 f 数组最后一个元素的个位是 0...9 的个数, 表示所有前 个元素的子序列中 0...9 的个数。转移显然。

#pragma GCC optimize(3)
#pragma GCC optimize("Ofast,no-stack-protector")
#include <bits/stdc++.h>
#define int long long
#define all(x) x.begin(),x.end()

using namespace std ;

const int inf = 0x3f3f3f3f ;
const int LLinf = 0x3f3f3f3f3f3f3f3f ;
const int mod = 1e9 + 7 ;
const int N = 1e5 + 10 ;

int f[N][10] , d[N][10] , n , a[N] ;

void solve()
{
	cin >> n ;
	for(int i = 1 ; i <= n ; ++ i)
	{
		cin >> a[i] ;
		a[i] %= 10 ;
	}
	for(int i = 1 ; i <= n ; ++ i)
	{
		for(int j = 0 ; j <= 9 ; ++ j)
		{
			d[i][j] = d[i - 1][j] * 2 % mod ;
			f[i][j] = f[i - 1][j] ;
		}
		f[i][a[i]] ++ ; 
		d[i][a[i]] ++ ;
		for(int j = 0 ; j <= 9 ; ++ j)
		{
			int t = (j * a[i]) % 10 ;
			f[i][t] = (f[i][t] + f[i - 1][j]) % mod ;
			d[i][t] = (d[i][t] + f[i - 1][j]) % mod ;
		}
	}
	cout << d[n][6] << '\n' ;
}

signed main()
{
	cin.tie(nullptr)->ios::sync_with_stdio(false) ;

	int t = 1 ;
	// cin >> t ;

	for(int i = 1 ; i <= t ; i++)
	{
		// cout << "case #" << i << " : " << '\n' ;
		solve() ; 
	}
	return 0 ; 
}
全部评论

相关推荐

11-28 17:48
中山大学 C++
点赞 评论 收藏
分享
听说改名字就能收到offer哈:Radis写错了兄弟
点赞 评论 收藏
分享
评论
2
收藏
分享
牛客网
牛客企业服务