子段乘积(尺取法+逆元 / 线段树)


原题链接:https://ac.nowcoder.com/acm/contest/3005/C

思路: 本题可以使用线段树,也可以使用尺取法 + 逆元。

  1. 尺取法+逆元:l 代表左端点,r 代表右端点。l 先不动,r 往前扫描,如果成功扫到,有 k 个非0元素的子段就累乘起来,最后把最左端的元素除了(用乘法逆元,否则会出现除以 0 的异常),左端点往前移动,l++,再继续扫描。在未达到 k 个非零元素的子段前,如果遇到 0,当前的区间重置 ,左端点直接到 0 的下一个位置继续扫描。
  2. 线段树:线段树裸题,只用到查询,还不用更新。

Code1:

//尺取法 + 乘法逆元
#include <iostream>
using namespace std;
typedef long long ll;
const int N=2e5+100;
const ll mod=998244353;
int n,k;
ll a[N];
ll q_pow(ll a,ll b){
    ll ans=1,res=a%mod;
    while(b){
        if(b&1)    ans=ans*res%mod;
        res=res*res%mod;
        b>>=1;
    }
    return ans%mod;
}
ll inv(ll a,ll mod){
    return q_pow(a,mod-2)%mod;
}
int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    int l=1,r=1;
    ll mul=1,ans=0;
    while(r<=n){
        if(a[r]){
            mul=mul*a[r]%mod;
            if(r-l+1==k){
                ans=max(ans,mul);
                mul=mul*inv(a[l],mod)%mod;
                l++;
            }
        }
        else{
            mul=1;
            l=r+1;
        }
        r++;
    }
    cout<<ans<<endl;
    return 0;
}

Code2

//线段树
#include <iostream>
using namespace std;
typedef long long ll;
const int N=2e5+100;
const ll mod = 998244353;
ll a[N];
struct node{
    int l,r;
    ll v;
}tree[N<<2];
void build_tree(int node,int l,int r){
    tree[node].l=l,    tree[node].r=r;
    if(l==r){
        tree[node].v=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build_tree(node<<1,l,mid);
    build_tree(node<<1|1,mid+1,r);
    tree[node].v=tree[node<<1].v*tree[node<<1|1].v%mod;
}
ll query(int node,int l,int r){
    if(tree[node].l==l && tree[node].r==r)
        return tree[node].v;
    int mid=(tree[node].l+tree[node].r)>>1;
    if(r<=mid)    return query(node<<1,l,r);
    if(l>mid)    return query(node<<1|1,l,r);
    return query(node<<1,l,mid)*query(node<<1|1,mid+1,r)%mod;
}
int main(){
    int n,k;    cin>>n>>k;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    build_tree(1,1,n);
    ll ans=0;
    for(int i=1;i<=n-k+1;i++)
        ans=max(ans,query(1,i,i+k-1));
    cout<<ans<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值