qdu校内排位赛 - G 搞算法要以和为贵 -（dp）

题目描述：

思路

一开始想的是dp [ i ] [ j ] :前 i 个数替换 j 个数的式子之和
但是看了看 n 到5000 , 内存限制32MB ，不大行
就优化了一下dp [ i ] ：n个数中替换了 i 个数的式子之和
dp [ j ] 由 dp [ j ] 和 dp [ j - 1 ] 得到，对于位置 i ,我们有两种取法，一个是取 ci，另一个是取 bi，
取 ci 由dp [ j ] * c [ i ] 得到；
取 bi , 由 dp [ j - 1 ] * b [ i ] 得到;
然后滚动一下，就得到答案了
负数 x 取模要（ x + mod）% mod
初始 b , c 数组也要取模，可能是负数

代码：

#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f
#define P pair<int,int>
#define lowbit(x) x & -x
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double eps=1e-7;
const double pai=acos(-1.0);
const int N=1e5+10;
const int maxn=1e5+10;
const int mod=998244353;
ll n,b[maxn],c[maxn],q,x,ans[maxn],dp[maxn];
void init()
{
	dp[0]=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=n;j>=0;j--)
		{
			dp[j]=(dp[j]*c[i])%mod;
			if(j>0) dp[j]=(dp[j]+dp[j-1]*b[i])%mod;
		}
    }
    for(int i=1;i<=n;i++)
    	dp[i]=(dp[i]+mod)%mod;
}
int main()
{
	scanf("%lld",&n);
	for(int i=1;i<=n;i++) scanf("%lld",&b[i]),b[i]=(b[i]+mod)%mod;
	for(int i=1;i<=n;i++) scanf("%lld",&c[i]),c[i]=(c[i]+mod)%mod;
	init(); 
	scanf("%lld",&q);
	while(q--)
	{
		scanf("%lld",&x);
		printf("%lld\n",dp[x]); 
	}
    return 0;
}