题意
给出由N个正整数组成的数组A,有Q次查询,每个查询包含一个整数K,从数组A中任选K个(K <= N)把他们乘在一起得到一个乘积。求所有不同的方案得到的乘积之和,由于结果巨大,输出Mod 100003的结果即可。例如:1 2 3,从中任选1个共3种方法,{1} {2} {3},和为6。从中任选2个共3种方法,{1 2} {1 3} {2 3},和为2 + 3 + 6 = 11。
1 <= N, Q <= 50000,1 <= A[i] <= 10^9
分析
没有想到分治。。。
可以把整个序列分成左右两边单独处理,然后合并的话就是一个卷积的形式,可以用NTT来搞。
但这里的模数并不是NTT模数。注意到每次卷积后的最大值在 1016 左右,不知道能不能用FFT来搞。这里可以用两个模数分别NTT,然后用天朝剩余定理合并就好了。
注意这里是每次NTT完之后都要合并一下,而不是全部搞完之后再合并。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=50005;
const int mod1=998244353;
const