埃森哲杯金马五校赛 L . K序列(dp)

最新推荐文章于 2020-01-31 21:42:57 发布

Werky_blog

最新推荐文章于 2020-01-31 21:42:57 发布

阅读量206

点赞数

分类专栏：简单DP

本文链接：https://blog.csdn.net/qq_33997572/article/details/79978895

版权

简单DP 专栏收录该内容

15 篇文章 1 订阅

订阅专栏

链接：https://www.nowcoder.com/acm/contest/91/L

来源：牛客网

题目描述

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

输入描述:

第一行为两个整数 n, K, 以空格分隔，第二行为 n 个整数，表示 a[1] ∼ a[n]，1 ≤ n ≤ 10⁵ , 1 ≤ a[i] ≤ 10⁹ , 1 ≤ nK ≤ 10⁷

输出描述:

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

dp[j]表示包括该点，序列和为j的最长子序列长度。

dp更新前，tmp[j]表示不包括该点，序列和为j的最长子序列长度，dp更新后跟新一下tmp便于下次更新。

初始化子序列和为0时tmp为1，其他为负无穷。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
const int inf=0x3f3f3f3f;

int a[maxn],dp[maxn],tmp[maxn];

int main(){
    int n,k;
    cin>>n>>k;
    memset(tmp,-inf,sizeof(tmp));
    tmp[0]=0;
    for(int i=0;i<n;i++)
        cin>>a[i];
    for(int i=0;i<n;i++){
        for(int j=0;j<k;j++){
            dp[(j+a[i])%k]=tmp[j]+1;
        }
        for(int j=0;j<k;j++){
            tmp[j]=max(tmp[j],dp[j]);
        }
    }
    printf("%d\n",dp[0]);
    return 0;
}