蓝桥杯 和与乘积 (前缀和)

链接：和与乘积

题意：
给定一个长度为n的数列，求有多少个区间 [L, R] 满足区间内元素的乘积等于他们的和。

思路：
首先我们知道乘积的增长比和要快很多，唯一比较特殊的就是当前数为1时，乘积不变，和单调增大。所以可以得出两个结论：
1.对于一个固定的起点，在一段连续的 1 中，至多只有一个位置满足要求。
2.当乘积超过一定的值时，后一段区间都不可能满足要求。
所以我们只需将不为1的位置存起来，对于每个左端点，遍历这些位置，同时判断这些位置之间的连续1是否存在一个接即可。因为乘积增长很快，我们最多遍历$log(n\ast n)$个位置。最后的时间复杂度为$n\ast log(4e10)$,可以通过本题。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 998244353;
const ll de = 4e10;
const int maxn = 2e6 + 7;
int n ;
vector<int> pos;
ll a[maxn] , pre[maxn];
int main(){
    scanf("%d",&n);
    pos.push_back(0);
    for(int i = 1; i <= n;i ++){
        scanf("%lld",&a[i]);
        if(a[i] != 1) pos.push_back(i);
        pre[i] = pre[i - 1] + a[i];
    }
    pos.push_back(n + 1);
    ll ans = 0;
    for(int l = 1; l <= n; l ++){
        ll res = 1;
        int now = lower_bound(pos.begin() , pos.end() , l) - pos.begin();//找到第一个大于等于l的不为1的位置
        for(int j = now; j < pos.size(); j ++){
            int cnt = pos[j] - pos[j - 1] - 1;
            int r = pos[j];
            if(res <= pre[r - 1] - pre[l - 1]){            //如果连续的1区间内有解
                if(pre[r - 1] - pre[l - 1] - res < cnt){   //连续1内有解
                    ans ++;
                }
            }
            res = res * a[r];
            if(r != n + 1 && res == pre[r] - pre[l - 1]){  //当前不为1的位置 是一个解
                ans ++;
            }
            if(res >= de) break;
        }
    }
    printf ("%lld\n",ans);
}