埃森哲杯金马五校赛 L K序列

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_37158899/article/details/79952192
链接:https://www.nowcoder.com/acm/contest/91/L
来源:牛客网

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 131072K,其他语言262144K
64bit IO Format: %lld

题目描述

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

输入描述:

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

输出描述:

输出一个整数表示最长子序列的长度 m
#include <bits/stdc++.h>
#include <cstring>

using namespace std;

long long sum[100010];
int main()
{
    int n,k,x;
    scanf("%d%d",&n,&k);
    memset(sum,0,sizeof(sum));
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&x);
        sum[i]=sum[i-1]+x;
    }
    int imax=0;
    int cha;
    for(int i = 1; i <= n; i++)
        for(int j = n; j > i; j--)
        {
            if((sum[j]-sum[i-1])%k==0)
            {
                imax = max(imax,j-i+1);
                break;
            }
        }
    printf("%d\n",imax);
    return 0;
}

前缀和求出来倒着算即可。

这样求出来的一定是最大的,同时减少时间复杂度。

但这其实是子串只是连续的。

不连续的子序列要用dp。

不会啊。

好了,我现在会了。

dp[j]表示到目前为止的数的和模k为j的子序列的最大长度。

然后只要从从第一个元素扫到最后一个元素即可。

注意边界条件:第一个元素单独处理,即初始化dp[a[0]%k]=1,其他都置0

状态转移方程为dp[(a[i]+j]%k] = max(dp[(a[i]+j]%k],dp[j]+1)

但这里要注意两点,

第一,因为每个值在添加一个新元素时只能更新一次,所以要用一个flag变量滚动以防数组被“污染”

第二,如果dp[(a[i]+j]%k]要被更新为dp[j]+1的条件是dp[j]必须不为0。因为如果为0意味着还没有数被添加进去,其和模j其实并不成立。

所以dp[j]为0时,dp[(j+a[i])%k]等于原值(上一次计算得到的结果)

本来我是没想明白为什么dp[j]为0也需要更新的。

然后我就把dp数组打印出来看看有什么区别。

左边是不更新,右边是更新;输入是题目给的样例,可以自己感受一下。


//#include <bits/stdc++.h>
#include <iostream>
#include <stdlib.h>
#include <algorithm>
#include <queue>
#include <stdio.h>
#include <string.h>
#include <vector>
#include <cmath>
#define max(a,b) (a>b)?a:b
const int N = 10;
int n,k;
int a[N];
int ans;
int dp[N][2];
int tmp[N];
int main()
{
   // memset(dp,-1,sizeof(dp));
    scanf("%d%d",&n,&k);
    for(int i = 0; i < n; i++)
    {
        scanf("%d",&a[i]);
        a[i] %= k;
    }

    int flag = 0;
    dp[a[0]%k][flag] = 1;
    for(int i = 1; i < n; i++)
    {
        flag = !flag;//滚动使用数组
        for(int j = 0; j < k; j++)
        {
            if(dp[j][!flag])
            {
                dp[(j+a[i])%k][flag] = max(dp[j][!flag]+1,dp[(j+a[i])%k][flag]);
            }
            else  dp[(j+a[i])%k][flag] =  dp[(j+a[i])%k][!flag];
        }
        }

    }
    printf("%d\n",dp[0][flag]);
    return 0;
}


阅读更多
想对作者说点什么?

博主推荐

换一批

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