专题（三）数学与简单DP—— AcWing 1214. 波动数列

JohnnyLin00

于 2021-03-08 13:25:20 发布

阅读量128

点赞数

分类专栏： # 备战蓝桥杯文章标签：动态规划

本文链接：https://blog.csdn.net/weixin_44855907/article/details/114527474

版权

备战蓝桥杯专栏收录该内容

44 篇文章 24 订阅

订阅专栏

该博客主要介绍了如何使用Java动态规划方法解决AcWing在线编程平台上的第1214题——波动数列。通过分析题目描述和给出的代码，我们可以看到这是一个关于组合问题的动态规划求解实例。代码中定义了一个二维数组f来存储状态转移方程，并使用get_mod函数处理模运算，确保结果在指定范围内。最终，程序输出了满足条件的方案数。

摘要由CSDN通过智能技术生成

【题目描述】

AcWing 1214. 波动数列

【思路】

动态规划组合问题
在这里插入图片描述

在这里插入图片描述

import java.util.Scanner;
public class Main{
    static int N = 1010;
    static int mod = 100000007;
    static int f[][] = new int [N][N];// f[i][j]表示前i个数且其和除以s余为j的方案的集合
    
    public static int get_mod(int x, int n){//可能为负数 将[- n + 1, 0]映射为[0, n - 1]
        return (x % n + n) % n;
    }
    public static void main(String args[]){
        Scanner reader = new Scanner(System.in);
        int n = reader.nextInt(), s = reader.nextInt(), a = reader.nextInt(), b = reader.nextInt();
        f[0][0] = 1;
        
        for(int i = 1; i < n; i ++)
            for(int j = 0; j < n; j ++)
            {
                f[i][j] = (f[i - 1][get_mod(j - a * i, n)] + f[i - 1][get_mod(j + b *i, n)])% mod;
            }
        System.out.println(f[n - 1][n - 1]);
    }
}