4 条题解

  • 1
    @ 2022-2-26 15:40:10

    #(〇)数学符号注释 本文中的某些符号体系并不标准,一些读者会在其他书中学习类似的内容。这里列出了他们可能不熟悉的符号。

    符号 名称 等价形式
    i=1nai\sum\limits_{i=1}^na_i 求和(和式) a1+a2++an1+anaa_1+a_2+\cdots+a_{n-1}+a_na
    i=1nai\prod\limits_{i=1}^na_i 求积 a1×a2××an1×anaa_1\times a_2\times\cdots\times a_{n-1}\times a_na
    [m=n][m=n] 如果m=nm=n值为11;否则为00 {1m=n0mn\begin{cases}1&m=n\\ 0& m\not=n \end{cases}

    (一)题目大意

    简化一下题意:

    nn个数(c1,c2,...,cn)(c_1,c_2,...,c_n)0ciai0\leqslant c_i\leqslant a_i ,求有多少种方案数使i=1nci=m\sum\limits_{i=1}^nc_i = m

    (二)解题思路

    乍一看,似乎题目有些复杂,一时找不到思路,肿么办!!!

    方法一:搜索

    没有思路当然就搜索啦 废话。如何搜索呢?

    11nn考虑每个 cic_i的值,和当前前 ii个数的总和 kk,然后枚举当前 xix_i 所有可能的值,再递归求解。

    时间复杂度 O(i=1nai)O(\prod\limits_{i=1}^na_i),明显超时,但可以拿部分分(30)嘛...

    代码:

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn=105, mod = 1000007;
    int n, m, a[maxn];
    int dfs(int x,int k)
    {
        if(k > m) return 0;
        if(k == m) return 1;
        if(x == n+1) return 0;
        int ans = 0;
        for(int i=0; i<=a[x]; i++) ans = (ans + dfs(x+1, k+i))%mod;
        return ans;
    }
    int main()
    {
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
        cout<<dfs(1,0)<<endl;
        return 0;
    }
    

    搜索超时怎么办!!! 别着急...

    方法二(搜索优化法宝):记忆化

    时间复杂度大概是:O(nmai)O(nma_i) 吧,100100%的数据稳过。

    代码(其实只是改动了一点点):

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn=105, mod = 1000007;
    int n, m, a[maxn], rmb[maxn][maxn];
    int dfs(int x,int k)
    {
        if(k > m) return 0;
        if(k == m) return 1;
        if(x == n+1) return 0;
        if(rmb[x][k]) return rmb[x][k]; //搜过了就返回
        int ans = 0;
        for(int i=0; i<=a[x]; i++) ans = (ans + dfs(x+1, k+i))%mod;
        rmb[x][k] = ans; //记录当前状态的结果
        return ans;
    }
    int main()
    {
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
        cout<<dfs(1,0)<<endl;
        return 0;
    }
    

    但是搜索的时间有些不稳定,想要更稳定的算法有木有...

    方法三:动态规划

    记忆化搜索都可以转成动态规划,但是动态规划却不一定能转成记忆化搜索 ——byclgby clg 定义状态:f(i,j)f(i, j) 表示前 ii 个数总和为 jj 的方案数。 那么,易得状态转移方程:f(i,j)=k=0aif(i1,jk)f(i, j) = \sum\limits_{k=0}^{a_{i}}f(i-1,j-k) 其中, kk是枚举当前第 ii 个数的取值。

    时间复杂度:O(nmai)O(nma_i),稳得一批。 空间复杂度:O(nm)O(nm)

    代码:

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn=105, mod = 1000007;
    int n, m, a[maxn], f[maxn][maxn];
    int main()
    {
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
        f[0][0] = 1;
        for(int i=1; i<=n; i++)
           for(int j=0; j<=m; j++)
               for(int k=0; k<=min(j, a[i]); k++)
                  f[i][j] = (f[i][j] + f[i-1][j-k])%mod;
        cout<<f[n][m]<<endl;
        return 0;
    }
    

    仔细观察上述代码,有木有发现什么...

    方法四(dp优化法宝):滚动数组

    因为我们发现,状态转移方程中,当前状态 f(i,j)f(i, j)只跟 f(i1,j)f(i-1, j) 有关系,与 i2,i3...i-2,i-3...无关。于是,我们可以利用滚动数组优化dp。

    所谓滚动数组,其实就是只保留两个状态(当前状态和前一个状态),算完当前状态后,将当前状态变为前一个状态,再去算下一个状态,看上去就像二维数组的两层不断地迭代。

    时间复杂度:O(nmai)O(nma_i) 空间复杂度:O(m)O(m)

    代码:

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn=105, mod = 1000007;
    int n, m, a[maxn], f[2][maxn], t;
    int main()
    {
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
        f[0][0] = 1;
        for(int i=1; i<=n; i++)
        {
            t = 1-t; //滚动
            for(int j=0; j<=m; j++)
            {
                f[t][j] = 0; //注意初始化
                for(int k=0; k<=min(j, a[i]); k++)
                  f[t][j] = (f[t][j] + f[1-t][j-k])%mod;
            }
        }
        cout<<f[t][m]<<endl;
        return 0;
    }
    

    看到上述dp代码,有木有感觉很熟悉... 这熟悉的优化方法... 这不就是个背包吗!!!

    方法五(背包大法好):一维动态规划

    通过观察上面的代码,二维数组,数组滚动优化空间......还有那熟悉的格式... 猛然发现这怎么可能不是背包呢(01背包)? 既然是背包,那么就可以为所欲为啦... [邪恶.jpg] 直接压成一维的01背包,跑一波,搞掂!!!

    时间复杂度:O(nmai)O(nma_i) 空间复杂度:O(m)O(m)

    代码:

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn=105, mod = 1000007;
    int n, m, a[maxn], f[maxn];
    int main()
    {
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
        f[0] = 1;
        for(int i=1; i<=n; i++)
            for(int j=m; j>=0; j--) //注意,是01背包
                for(int k=1; k<=min(a[i], j); k++)
                  f[j] = (f[j] + f[j-k])%mod;
        cout<<f[m]<<endl;
        return 0;
    }
    

    方法六:前缀和优化

    继续观察方法五的代码,时间复杂度是Θ(n3)\Theta(n^3))级别的。与背包有一些差别的是这一句:

    for(int k=1; k<=min(a[i], j); k++)
        f[j] = (f[j] + f[j-k])%mod;
    

    然而,这一句的作用只不过是将连续的一段f[ja[i]]f[j-a[i]]f[j1]f[j-1]累加起来而已。因此,可以用前缀和将这个操作优化(众所周知,前缀和的作用就是Θ(1)\Theta(1)求一段区间的和)。

    时间复杂度:Θ(nm)\Theta(nm)

    顺便卡到了次优解。

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn = 105, mod = 1000007;
    int n, m, f[maxn], sum[maxn], a[maxn];
    int main(){
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
    	f[0] = 1;
        for(int i=0; i<=m; i++) sum[i] = 1;
        for(int i=1; i<=n; i++){
        	for(int j=m; j>=1; j--) f[j] = (f[j] + sum[j-1] - sum[j - min(a[i], j) - 1] + mod)%mod;
    		for(int j=1; j<=m; j++) sum[j] = (sum[j-1] + f[j])%mod;
    	}
        cout<<f[m]<<endl;
        return 0;
    }
    

    提示:上面的程序在计算f[j]f[j]的时候有可能会出现数组越界的情况,但为了代码美观容易理解,没有特判。这一点需要注意,考场上不慎就会抱灵。

    下面是加上了特判的代码:

    #include<bits/stdc++.h>
    using namespace std;
    const int maxn = 105, mod = 1000007;
    int n, m, f[maxn], sum[maxn], a[maxn];
    int main(){
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
    	f[0] = 1;
        for(int i=0; i<=m; i++) sum[i] = 1;
        for(int i=1; i<=n; i++){
        	for(int j=m; j>=1; j--){
        		int t = j - min(a[i], j) - 1;
        		if(t < 0) f[j] = (f[j] + sum[j-1])%mod;
        		else f[j] = (f[j] + sum[j-1] - sum[t] + mod)%mod;
    		}
    		for(int j=1; j<=m; j++) sum[j] = (sum[j-1] + f[j])%mod;
    	}
        cout<<f[m]<<endl;
        return 0;
    }
    

    方法七:生成函数

    建议初学者跳过此方法,权当提供一种思路。

    回到开头,我们需要求的是这样一个式子:

    ck=0ak[i=1nci=m],k={1,2,...,n1,n}\sum\limits_{c_k=0}^{a_k}\left[\sum\limits_{i=1}^nc_i=m\right ] ,k=\{1,2,...,n-1,n\}

    于是我们可以构造这样一个生成函数:

    g(x)=(1+x1+x2++xa1)(1+x1+x2++xa2)(1+x1+x2++xan)g(x)=(1+x^1+x^2+\cdots+x^{a_1})(1+x^1+x^2+\cdots+x^{a_2})\cdots(1+x^1+x^2+\cdots+x^{a_n}) 即,

    g(x)=i=1nj=0aixjg(x)=\prod\limits_{i=1}^n\sum_{j=0}^{a_i}x^j

    7.1多项式

    最后所得的多项式中,xmx^m项的系数即为答案。可以做nn次NTT得到答案。假设aia_i的值域是kk ,那么, 时间复杂度:Θ(n2klognk)\Theta(n^2k\log nk) 实际上,mm最大可以取到nknk。因此如果用kk来表示,前缀和优化后的复杂度应当是:Θ(n2k)\Theta(n^2k) NTT的局限在于要做nn次。其实只需要取ln\ln就能将多项乘法转化为多项式加法,具体参考付公主的背包。 时间复杂度降到了Θ(nklognk)\Theta(nk\log nk),即 Θ(mlogm)\Theta(m\log m) ###7.2封闭形式 也可以将其写成封闭形式,参考[CEOI2004] Sweets。 等比数列求和,得到:

    g(x)=i=1n(1xai)(1x)ng(x)=\dfrac{\prod_{i=1}^n(1-x^{a_i})}{(1-x)^n}

    nn 较小,mm 很大的时候,可以考虑将 i=1n(1xai)\prod_{i=1}^n(1-x^{a_i}) 暴力展开,对于 (1x)n(1x)(1-x)^{-n}(1−x) 使用牛顿二项式定理。即,

    (1x)n=i0(n+i1i)xi(1-x)^{-n}=\sum_{i\ge0}\binom {n+i-1}{i}x^i

    枚举前一个式子 xx 的幂次,假设为 kk,设 xkx^k项的系数为 ckc_k ,那么,

    ANS=k=0mck(n+mk1mk){\rm {ANS}}=\sum _{k=0}^m c_k\cdot\binom {n+m-k-1}{m-k}

    时间复杂度:Θ(2n+m)\Theta(2^n+m)

    (三)总结

    总的来说,这道题适合 搜索/动态规划 的初学者练习。 有一点点的思维难度。 这道题的价值在于,它既可以从简单的动态规划开始,一路优化,也可以从生成函数的视角观察,继续优化。这两条道路,竟是殊途同归。或许,这也是数学的魅力吧。

    摘自洛谷P1077 [NOIP2012 普及组] 摆花 题解的第一个题解,删除了其中的肮脏语言

    • 0
      @ 2023-5-26 19:46:47

      #include <bits/stdc++.h> #define int long long using namespace std; int n,m,a[110],f[110][110],mod=1000007; signed main() { cin>>n>>m; for(int i=1;i<=n;i++) cin>>a[i]; f[0][0]=1; for(int i=1;i<=n;i++) for(int j=0;j<=m;j++) for(int k=0;k<=min(a[i],j);k++) f[i][j]=(f[i][j]+f[i-1][j-k])%mod; cout<<f[n][m]; return 0; }

      • 0
        @ 2023-5-26 19:42:12

        #include <bits/stdc++.h> #define int long long using namespace std; int n,m,a[110],f[110][110],mod=1000007; signed main() { cin>>n>>m; for(int i=1;i<=n;i++) cin>>a[i]; f[0][0]=1; for(int i=1;i<=n;i++) for(int j=0;j<=m;j++) for(int k=0;k<=min(a[i],j);k++) f[i][j]=(f[i][j]+f[i-1][j-k])%mod; cout<<f[n][m]; return 0; }

        • 0
          @ 2021-10-6 19:52:44
          #include<bits/stdc++.h>
          using namespace std;
          int n,m;
          int dp[1010],a[1010];
          int main(){
          	cin>>n>>m;
          	for(int i=1;i<=n;i++) cin>>a[i];
              dp[0]=1;
          	for(int i=1;i<=n;i++){
          		for(int j=m;j>=0;j--){
          			for(int k=1;k<=a[i];k++){
          				if(j>=k&&dp[j-k]>0) dp[j]=(dp[j]+dp[j-k])%1000007;
          
          			    }
          	    }
              }
              cout<<dp[m];
              return 0;
          }
          
          • 1

          信息

          ID
          724
          时间
          1000ms
          内存
          256MiB
          难度
          5
          标签
          递交数
          160
          已通过
          63
          上传者