长度为k的子序列

最新推荐文章于 2023-03-11 17:08:49 发布

星海

最新推荐文章于 2023-03-11 17:08:49 发布

阅读量369

点赞数

分类专栏：动态规划文章标签：动态规划

本文链接：https://blog.csdn.net/qq_41056867/article/details/124136680

版权

动态规划专栏收录该内容

6 篇文章 0 订阅

订阅专栏

这篇博客探讨了一种使用动态规划方法解决计算给定数组中长度为k的子序列中和为偶数的个数的问题。程序采用Java实现，通过维护两个状态的动态规划数组来分别记录和为偶数和奇数的子序列个数。核心算法是根据当前元素的奇偶性更新dp数组，最终得到答案。

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

长度为k的子序列中，和为偶数的个数。

import java.util.*;
public class Main {
    public static void main(String[] args){
        Scanner in=new Scanner(System.in);
        int t=in.nextInt();
        for(int i=0;i<t;i++){
            int n=in.nextInt();
            int k=in.nextInt();
            int[] nums=new int[n];
            for(int j=0;j<n;j++){
                nums[j]=in.nextInt();
            }
            System.out.println(f(nums,k));
        }
    }
    private static int f(int[] nums,int k){
        int ans=0;
        int mod=10^9+7;
        int[][] dp=new int[k][2];//dp[j][0]长度为j的序列中和为偶数的个数，dp[j][1]长度为j的序列中和为奇数的个数
        dp[0][0]=1;
        dp[0][1]=0;
        for(int i=0;i<nums.length;i++){
            if(nums[i]%2==0){
                ans=ans+dp[k-1][0];
            }else{
                ans+=dp[k-1][1];
            }
            ans%=mod;
            for(int j=k-1;j>=1;j--){
                if(nums[i]%2==0) {
                    dp[j][0]+=dp[j-1][0];
                    dp[j][1]+=dp[j-1][1];
                }else{
                    dp[j][0]+=dp[j-1][1];
                    dp[j][1]+=dp[j-1][0];
                }
                dp[j][0]%=mod;
                dp[j][1]%=mod;
            }
        }
        return ans;
    }
}