序列区间
题目链接:luogu U137902
题目大意
给你一个序列,问你有多少个子区间,满足子区间数的积除子区间数的和是给出的值 k。
思路
首先化式子:
∏
i
=
l
r
a
i
∑
i
=
l
r
a
i
=
k
\dfrac{\prod\limits_{i=l}^ra_i}{\sum\limits_{i=l}^ra_i}=k
i=l∑raii=l∏rai=k
∏ i = l r a i = ∑ i = l r a i k \prod\limits_{i=l}^ra_i=\sum\limits_{i=l}^ra_ik i=l∏rai=i=l∑raik
你考虑看
80
%
80\%
80% 的数据说数都大于
1
1
1。
那你想左边的数最大你算一下就只有
2
e
18
2e18
2e18,那左边的肯定也不能大于
18
18
18。
那你乘起来如果都不是 1 1 1 每次至少翻一倍,那最多也只能乘 60 60 60 次,那我们暴力的时候判断一下,就降到 O ( n l o g 1 0 18 ) O(nlog10^{18}) O(nlog1018) 了。
接着考虑单独处理
1
1
1。
考虑放一个
1
1
1 会改变什么,发现左边不会变,右边会加
k
k
k。
那你可以处理出连续的
1
1
1,那每次就是加上
k
k
k,相当于
+
k
,
+
2
k
,
.
.
.
,
+
i
k
+k,+2k,...,+ik
+k,+2k,...,+ik(
i
i
i 是连续的个数)
然后这个之中有等于左边的就加一下贡献,然后复杂度就变回去了。
代码
#include<cstdio>
#define ll long long
using namespace std;
ll n, k, a[200001], maxn;
int r[200001], to1[200001];
ll up, down, ans;
int main() {
scanf("%lld %lld", &n, &k);
for (int i = 1; i <= n; i++) scanf("%lld", &a[i]), maxn += a[i];
maxn *= k;
int now = 1; ll nowx = 1;
for (int i = 1; i <= n; i++) {
while (now <= n && (long double)nowx * (long double)a[now] <= (long double)maxn) {
nowx *= a[now]; now++;
}
r[i] = now - 1;
if (now > i) {
nowx /= a[i];
}
}
for (int i = n; i >= 1; i--)
if (a[i] == 1) {
if (a[i + 1] == 1) {
to1[i] = to1[i + 1];
}
else to1[i] = i;
}
for (int i = 1; i <= n; i++) {
down = 0; up = 1;
for (int j = i; j <= r[i]; j++) {
if (a[j] == 1) {
int dis = to1[j] - j + 1;
if (down < up && down + k * dis >= up && (up - down) % k == 0) ans++;
down += k * dis;
j = to1[j];
}
else {
down += k * a[j];
up *= a[j];
if (up == down) ans++;
}
}
}
printf("%lld", ans);
return 0;
}