题目描述 :
本题分为简单版本和困难版本,二者唯一的区别是:简单版本有序列 a a a 所有元素乘积 ≤ 1 0 18 \leq 10^{18} ≤1018 的限制,困难版本没有。
氧气少年最近喜欢上了零。
给出一个长度为 n ( 1 ≤ n ≤ 2 ⋅ 1 0 5 ) n(1\leq n\leq 2\cdot 10^5) n(1≤n≤2⋅105) 的序列 a ( 1 ≤ a i ≤ 1 0 9 ) a(1\leq a_i\leq 10^9) a(1≤ai≤109),并且保证序列 a a a 所有元素乘积 ≤ 1 0 18 \leq 10^{18} ≤1018 ,求这个序列中满足如下条件的连续子段 [ a l … a r ] [a_l\dots a_r] [al…ar] 的数量:
- 令 x = a l ⋅ a l + 1 ⋅ a l + 2 … a r x=a_l\cdot a_{l+1}\cdot a_{l+2}\dots a_r x=al⋅al+1⋅al+2…ar,那么 x x x 的末尾恰好有 k ( 0 ≤ k ≤ 1 0 9 ) k(0\leq k\leq 10^9) k(0≤k≤109) 个零。
输入描述:
第一行包含一个整数 T ( 1 ≤ T ≤ 1 0 5 ) T(1\leq T\leq 10^5) T(1≤T≤105),表示测试用例的组数。
对于每组测试用例:
第一行包含两个整数 n ( 1 ≤ n ≤ 2 ⋅ 1 0 5 ) n(1\leq n\leq 2\cdot 10^5) n(1≤n≤2⋅105) 和 k ( 0 ≤ k ≤ 1 0 9 ) k(0\leq k\leq 10^9) k(0≤k≤109) ,表示序列的长度和题目中提到的后导零的数量;
第二行包含 n n n 个整数 a 1 … a n ( 1 ≤ a i ≤ 1 0 9 ) a_1\dots a_n\ (1\leq a_i\leq 10^9) a1…an (1≤ai≤109),表示该序列。
保证对于所有的测试用例, n n n 的总和不超过 2 ⋅ 1 0 5 2\cdot 10^5 2⋅105。
输出描述:
对于每组测试用例:
仅输出一行,包含一个整数,表示答案。
示例1
输入2 5 3 125 1 8 1 1 1 0 6
输出
3 1
题目大意:
找出符合连续积之后的后缀 0 0 0 个数为 k k k 的连续子段个数。
(例: (125, 1, 8)、(125, 1, 8, 1) 和 (125, 1, 8, 1, 1) 这 3 个连续积都为1000,然后后缀 0 个数都为 3,所以符合要求的连续子段个数为 3 )
解题思路:
由 2 ∗ 5 = 10 2*5=10 2∗5=10,可得后缀 0 0 0 个数为 因子2 和 因子5 的对数的数量(即一对2和5产生一个0)。那么,接下来就是求出每个数的 因子2和5 的个数。为了后面更好地知道一段连续子段的 因子2和5 的个数,那么我们采用前缀和分别记录前 n n n 个 因子2和5 的个数。
然后我们要快速的找到符合要求的区间,那么首先我们就得知道什么才算是符合要求的,由上可得,前缀和对应下标 因子2和5 的个数中最小的那一个与 k k k 相同(即因子2和5的对数为 k k k )就是符合要求的。
知道什么是符合要求的下标后,我们就得找区间边界了,因为知道边界就可以知道符合要求的连续知道个数了。这里,我们可以考虑枚举在左下标之后需要因子2(因子5)的个数(即前一个下标需要因子2(因子5)的个数 + k k k) ,然后通过上面得出的关系得出左右边界。二分分别查找 因子2和5 中符合要求的区间,然后这两个区间中,它们两个对应值较小要符合要求的话,可以观察出最后符合要求的左右边界为它们左右边界各自较大的那个边界(左边界为它们左边界较大的那个,右边界为它们左边界较大的那个)。
最后把每次枚举得出的边界相减的结果累加就是符合要求的连续子段个数。
#include <bits/stdc++.h>
using namespace std;
const int N =2e5+10;
int s2[N], s5[N]; //记录 因子2和因子5 的前 n 项个数和
int main()
{
s2[0] = s5[0] = 0;
int t; cin>>t;
while(t--)
{
int n, k; cin>>n>>k;
for(int i=0; i<=n; i++) s2[i] = s5[i] = 0;
for(int i=1; i<=n; i++)
{
int val; cin>>val;
//求当前值因子2的个数 + 之前所有值因子2的个数
s2[i] = s2[i-1];
while(val%2 == 0)
{
s2[i]++;
val /= 2;
}
//求当前值因子5的个数 + 之前所有值因子5的个数
s5[i] = s5[i-1];
while(val%5 == 0)
{
s5[i]++;
val /= 5;
}
}
long long cnt = 0;
for(int i=1; i<=n; i++)
{
int p2 = s2[i-1] + k; //找出在 i 之后需要的 因子2 的个数
int p5 = s5[i-1] + k; //找出在 i 之后需要的 因子5 的个数
//找出i之后所需因子2个数的左闭区间 >=
int l2 = lower_bound(s2, s2+1+n, p2) - s2;
//找出i之后所需因子2个数的右开区间 >
int r2 = upper_bound(s2, s2+1+n, p2) - s2;
//找出i之后所需因子5个数同理
int l5 = lower_bound(s5, s5+1+n, p5) - s5;
int r5 = upper_bound(s5, s5+1+n, p5) - s5;
//分别找出左区间的较大者,右区间的较大者,构成的新区间才是符合符合要求的
int l = max(l2, l5), r = max(r2, r5);
//因为是从 i 开始枚举的,所以 l 比 i 小就从 i 开始
l = max(i, l);
//如果不存在这样的区间 l 和 r 会相等,都为 n+1,相减不影响最后结果
cnt += r-l;
}
cout<<cnt<<'\n';
}
return 0;
}