动态规划 || 满足条件不连续子序列长度(两个数组存放结果节约空间)

题目描述 
给一个数组 a,长度为 n,若某个子序列中的和为 K 的倍数,那么这个序列被称为“K 序列”。现在要你 对数组 a 求出最长的子序列的长度,满足这个序列是 K 序列。 

输入描述:
第一行为两个整数 n, K, 以空格分隔,第二行为 n 个整数,表示 a[1] ∼ a[n],1 ≤ n ≤ 105 , 1 ≤ a[i] ≤ 109 , 1 ≤ nK ≤ 107

输出描述:
输出一个整数表示最长子序列的长度 m

输入
7 5
10 3 4 2 2 9 8

输出
6

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<math.h>
#include<vector>
#include<string>
#include<string.h>
#include<queue>
using namespace std;
typedef long long ll;
int k,n;
int a[100005];
int dp[2][10000005];
int main(){
    int ans=0;
    scanf("%d%d",&n,&k);
    int b;
    int m=0;
    for(int i=0;i<n;i++){
        scanf("%d",&b);
        if(b%k==0)ans++;
        else{
            a[m]=b%k;
            m++;
        }
    }
    if(m==0){
    	printf("%d\n",ans);
    	return 0;
    }//!!!
    memset(dp,0,sizeof(dp));
    dp[0][a[0]]=1;
    bool f=0;
    for(int i=1;i<m;i++){
    	f=i%2;
        for(int j=0;j<k;j++){
            if(dp[!f][((j-a[i]+k)%k)]==0)dp[f][j]=dp[!f][j];
            else dp[f][j]=max(dp[!f][((j-a[i]+k)%k)]+1,dp[!f][j]);
        }
    }
    printf("%d\n",ans+dp[f][0]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/bekote/article/details/80006814
今日推荐