牛客3005C-子段乘积-线段树

链接：https://ac.nowcoder.com/acm/contest/3005/C
来源：牛客网

题目描述：

给出一个长度为 n 的数列 a1,a2,…,an，求其长度为 k 的连续子段的乘积对 998244353 取模余数的最大值。

输入描述：

第一行两个整数n,k。
第二行n个整数,a1,a2,…,an​。

输出描述：

输出一个整数，代表最大余数。

输入样例：

5 3
1 2 3 0 8

输出样例：

6

核心思想：

线段树，建树时赋值，无更新，n-k+1次区间查询。

代码如下：

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=2e5+20;
const ll mo=998244353;
struct node{
	int l,r;
	ll v;
}tr[N<<2];
void pushup(int m)
{
	tr[m].v=tr[m<<1].v*tr[m<<1|1].v%mo;
	return;
}
void build(int m,int l,int r)
{
	tr[m].l=l;
	tr[m].r=r;
	if(l==r)
	{
		scanf("%lld",&tr[m].v);
		return;
	}
	int mid=(l+r)>>1;
	build(m<<1,l,mid);
	build(m<<1|1,mid+1,r);
	pushup(m);
	return;
}
ll query(int m,int l,int r)
{
	if(tr[m].l==l&&tr[m].r==r)
		return tr[m].v;
	int mid=(tr[m].l+tr[m].r)>>1;
	if(r<=mid)
		return query(m<<1,l,r);
	if(l>mid)
		return query(m<<1|1,l,r);
	return query(m<<1,l,mid)*query(m<<1|1,mid+1,r)%mo;
}
int main()
{
	int n,k;
	cin>>n>>k;
	build(1,1,n);
	ll ans=0;
	for(int i=1;i+k-1<=n;i++)
		ans=max(ans,query(1,i,i+k-1));
	cout<<ans<<endl;
	return 0;
}