思路: 本題可以使用線段樹,也可以使用尺取法 + 逆元。
- 尺取法+逆元:l 代表左端點,r 代表右端點。l 先不動,r 往前掃描,如果成功掃到,有 k 個非0元素的子段就累乘起來,最後把最左端的元素除了(用乘法逆元,否則會出現除以 0 的異常),左端點往前移動,l++,再繼續掃描。在未達到 k 個非零元素的子段前,如果遇到 0,當前的區間重置 ,左端點直接到 0 的下一個位置繼續掃描。
- 線段樹:線段樹裸題,只用到查詢,還不用更新。
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;
}