【牛客网】【埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛】L—K序列【动态规划】

题目链接:L—K序列

题解:for一遍数组,temp[i]表示:不包括当前for到的点,序列和为i的最长长度。然后for一遍0-k,对dp[ (j+num[i])%k ]进行更新,dp[j]是包括当前点的,序列和为j的最长长度。更新完dp之后再更新temp。复杂度是n*k,题目里说明了这个小于10的7次方。

#include<iostream>
#include<stdio.h>
#include<string>
#include<string.h>
#include<algorithm>
#include<vector>
#include<stack>
#include<queue>
#include<set>
#include<map>
using namespace std;
int num[100000+10];
int temp[10000000+10];
int dp[10000000+10];
int main()
{
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;i++){
        scanf("%d",&num[i]);
	}
	for(int i=0;i<n;i++){
        for(int j=0;j<k;j++){
            if(j==0||temp[j]!=0)
                dp[(j+num[i])%k]=temp[j]+1;
        }
        for(int j=0;j<k;j++){
            temp[j]=max(temp[j],dp[j]);
        }
	}
	printf("%d\n",dp[0]);
}

猜你喜欢

转载自blog.csdn.net/gymgym1212/article/details/79951341