牛客小白月赛73 D/E-Kevin喜欢零(简单版本/困难版本)

Kevin喜欢零(简单版本)
Kevin喜欢零(困难版本)

题目描述:
本题分为简单版本和困难版本，二者唯一的区别是：简单版本有序列 a 所有元素乘积 $\le$ $10^8$ 的限制，困难版本没有。

氧气少年最近喜欢上了零。

给出一个长度为 n(1 $\le$ n $\le$ 2 $\cdot$ $10^5$)的序列 a(1 $\le$ $a_i\le$ $10^9$)，并且保证序列 a 所有元素乘积 $\le$ $10^{18}$，求这个序列中满足如下条件的连续字段[$a_l$$\cdot$$\cdot$$\cdot$$a_r$]的数量：

令 x = $a_l$$\cdot$$a_{l+1}$$\cdot$$a_{l+2}$…$a_r$，那么 x 的末尾恰有有 k(0 $\le$ k $\le$ $10^9$) 个零。

输入描述:

第一行包含一个整数 T(1 $\le$ T $\le$ $10^5$)，表示测试用例的组数。

对于每组测试用例：

第一行包含两个整数 n(1 $\le$ T $\le$ 2 $\cdot$ $10^5$) 和 k(1 $\le$ k $\le$ $10^9$)，表示序列的长度和题目中提到的后导零的数量；

第二行包含 n 个整数 $a_1$…$a_n$(1 $\le$ $a_i$ $\le$ $10^9$)，表示该序列。

保证对于所有的测试用例，n 的总和不超过 2 $\cdot$ $10^5$。

对于每组测试用例：

仅输出一行，包含一个整数，表示答案。

示例1
2
5 3
125 1 8 1 1
1 0
6
输出
3
1

题目大意：
给你一段连续的序列，求满足条件的连续的子序列数，条件就是连续的子序列相乘的后导零个数等于输入的k
解题思路：
综合产生后导零的两数相乘差不多有1x10,2x5,25x8,15x6等等，可以发现除了1乘本身有后导零的数以外，对后导零有贡献的均为一对由2的倍数和5的倍数组成，也就是说，因子2和5有贡献。不难想到对数一定是取2的因子数和5的因子数最小的那个。然后为了达到题目所求连续的子序列数，可以想到使用用前缀和维护某一段的关于2和5的因子数。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int N = 15000;
signed main()
{
    int t;cin>>t;
    while(t--){
        int n,k,val;cin>>n>>k;
        ll s2[n+1],s5[n+1],cnt=0;
        for(int i=0; i<=n; i++) s2[i]=s5[i]=0;
        for(int i=1; i<=n; i++){
            cin>>val;
            s2[i] = s2[i-1];
            while(val%2==0){
                s2[i]++;
                val/=2;
            }
            s5[i] = s5[i-1];
            while(val%5==0){
                s5[i]++;
                val/=5;
            }
        }
        for(int i=1; i<=n; i++){
            int a2 = s2[i-1]+k,a5 = s5[i-1]+k;
            int l2 = lower_bound(s2,s2+n+1,a2)-s2;
            int r2 = upper_bound(s2,s2+n+1,a2)-s2;
            int l5 = lower_bound(s5,s5+n+1,a5)-s5;
            int r5 = upper_bound(s5,s5+n+1,a5)-s5;
            int l = max(l2,l5),r = max(r2,r5);
            l = max(i,l);
            cnt += (r-l);
        }
        cout<<cnt<<"\n";
    }
    return 0;
}