【luogu U137902】序列区间（模拟）

序列区间

题目链接：luogu U137902

题目大意

给你一个序列，问你有多少个子区间，满足子区间数的积除子区间数的和是给出的值 k。

思路

首先化式子：
$\frac{\prod _{i=l}^r{a}_i}{\sum _{i=l}^r{a}_i}=k$

$\prod _{i=l}^r{a}_i=\sum _{i=l}^r{a}_{i}k$

你考虑看 $80\%$ 的数据说数都大于 $1$。
那你想左边的数最大你算一下就只有 $2e18$，那左边的肯定也不能大于 $18$。

那你乘起来如果都不是 $1$ 每次至少翻一倍，那最多也只能乘 $60$ 次，那我们暴力的时候判断一下，就降到 $O(nlog10^{18})$ 了。

接着考虑单独处理 $1$。
考虑放一个 $1$ 会改变什么，发现左边不会变，右边会加 $k$。

那你可以处理出连续的 $1$，那每次就是加上 $k$，相当于 $+k,+2k,...,+ik$（$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;
}