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;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值