洛谷P5087 数学

最新推荐文章于 2023-11-23 23:26:42 发布

Cr.rech

最新推荐文章于 2023-11-23 23:26:42 发布

阅读量190

点赞数

分类专栏：做题笔记学习笔记

本文链接：https://blog.csdn.net/bcr_233/article/details/87869771

版权

学习笔记同时被 2 个专栏收录

25 篇文章 0 订阅

订阅专栏

做题笔记

18 篇文章 0 订阅

订阅专栏

题目链接

思路：一个很容易想的DP

设 $f_{i,j}$ 表示在前 $i$ 个数中取 $j$ 个所得到的乘积。

则 $f_{i,j}=f_{i-1,j}+f_{i-1,j-1}×a_i \quad(i>0,j>0)$
边界： $f_{0,0}=1\quad f_{0,i}=0\ (i\ne0) \quad f_{i,0}=0\ (i\ne 0)$

但是直接DP会爆空间。。。
观察到DP时只需调用到 $f_{i-1}$ ，所以可以用滚动数组的方式优化空间。

代码时间复杂度 $\mathcal O(nk)$ ，空间复杂度 $\mathcal O(n+k)$

const
	mo=1000000007; //膜数
var
	n,k,i,j:longint;
	a:array[0..100010] of longint;
	f1,f2:array[0..500] of int64; //安全起见开int64
begin
	readln(n,k);
	for i:=1 to n do read(a[i]);
	f1[0]:=1; //边界初始化
	for i:=1 to n do //DP
	begin
		for j:=0 to k do
			f2[j]:=(f1[j]+f1[j-1]*a[i]) mod mo;
		f1:=f2;
	end;
	writeln(f1[k]);
end.

做完题后仔细想了一下，发现 $\rm a$ 数组的空间也是可以省掉的，在线处理就可以，不用存。
然后就有了空间复杂度 $\mathcal O(k) $ 的代码：

const mo=1000000007;
var
    n,k,i,j,a:longint;
    f1,f2:array[0..500] of int64;
begin
    readln(n,k);
    f1[0]:=1;
    for i:=1 to n do
    begin
        read(a); //这样就可以把a压成变量了QAQ
        for j:=0 to k do
            f2[j]:=(f1[j]+f1[j-1]*a) mod mo;
        f1:=f2;
    end;
    writeln(f1[k]);
end.