子段乘積(尺取法+逆元 / 線段樹)


原題鏈接: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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章