埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛L

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/becky_w/article/details/79950981

K序列
链接:https://www.nowcoder.com/acm/contest/91/L
来源:牛客网

题目描述
给一个数组 a,长度为 n,若某个子序列中的和为 K 的倍数,那么这个序列被称为“K 序列”。现在要你 对数组 a 求出最长的子序列的长度,满足这个序列是 K 序列。
输入描述:
第一行为两个整数 n, K, 以空格分隔,第二行为 n 个整数,表示 a[1] ∼ a[n],1 ≤ n ≤ 105 , 1 ≤ a[i] ≤ 109 , 1 ≤ nK ≤ 107
输出描述:
输出一个整数表示最长子序列的长度 m
示例1
输入
7 5
10 3 4 2 2 9 8
输出
6

题目分析:我们用一个数组mod[i] 存 和取模k余数为i的最长序列个数,初始化每个个数都为-inf,那么如果我们新增加一个数t,我们知道要先原来mod[i]中个数不为-inf,那么 令 m=(i+t)%k
此时 mod[m]=mod[i]+1 由于更新的时候,前面的更新值会对后面的值产生影响,所以,我们用一个tmp数组暂存一下

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e7+50;
int mod[maxn];
ll a[maxn];
int tmp[maxn];
const int inf=0x3f3f3f3f;
int n,k;
int main()
{
    int n;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
    }
    for(int i=0;i<k;i++)
    {    mod[i]=-inf;
         tmp[i]=-inf;
    }
    for(int i=1;i<=n;i++)
    {
        int t=a[i]%k;
        for(int j=0;j<k;j++)
        {
            int m=(j+t)%k;
           if(mod[j]>0)
           {
               tmp[m]=max(tmp[m],mod[j]+1);
           }
           else
           {
               if(j==t) tmp[t]=max(tmp[t],1);
           }
        }
        for(int j=0;j<k;j++)
        {
            mod[j]=tmp[j];
        }
    }
    printf("%d\n",mod[0]);
    return 0;
}
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭