两个题目一样数据量不同。
10的因子是2,5,也就是求2,5里面小的那个
先求出每个数有2,5的因子,再使用前缀和求出到每个位置有多少个2,5因子。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[200010];
int c2[200010],c5[200010];//2,5的因子数组
int p2[200010],p5[200010];//前缀和数组
int main()
{
int t,n,k;cin>>t;
while(t--)
{
cin>>n>>k;
for(int i=0;i<=n;i++) c2[i]=c5[i]=0;//初始化因子数组
for(int i=1;i<=n;i++)
{
cin>>a[i];
while(a[i]%2==0) {a[i]/=2;c2[i]++;}
while(a[i]%5==0) {a[i]/=5;c5[i]++;}
p2[i]=p2[i-1]+c2[i];//c2[i]的前缀和
p5[i]=p5[i-1]+c5[i];//c5[i]的前缀和
}
long long ans=0;
for(int i=1;i<=n;i++)
{
int v2=k+p2[i-1];
int v5=k+p5[i-1];
int l2= lower_bound(p2,p2+1+n,v2)-p2,r2= upper_bound(p2,p2+1+n,v2)-1-p2;//找出需要的值在前缀和数组里面占的长度
int l5= lower_bound(p5,p5+1+n,v5)-p5,r5= upper_bound(p5,p5+1+n,v5)-1-p5;
int l=max(l2,l5),r=max(r2,r5);
l=max(l,i);//防止l<i;
if(l==n+1) continue;//没有找到;
ans=ans+(r-l+1);
}
cout<<ans<<endl;
}
return 0;
}