题目描述
给一个数组 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;
}