K-序列(埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛)
题目描述
给一个数组 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
写的时候暴力过了,但标准解法是用dp来写的。二维K数组保存结果
#include <iostream> #include <algorithm> #include <cstring> #include <cstdio> #include <vector> #include <queue> #include <stack> #include <cstdlib> #include <iomanip> #include <cmath> #include <cassert> #include <ctime> #include <map> #include <set> using namespace std; #pragma comment(linker, "/stck:1024000000,1024000000") #define lowbit(x) (x&(-x)) #define max(x,y) (x>=y?x:y) #define min(x,y) (x<=y?x:y) #define MAX 100000000000000000 #define MOD 1000000007 #define pi acos(-1.0) #define ei exp(1) #define PI 3.1415926535897932384626433832 #define ios() ios::sync_with_stdio(true) #define INF 0x3f3f3f3f #define mem(a) (memset(a,0,sizeof(a))) #define ll long long int dp[10000005][2]; int a[100005],n,k; int main() { scanf("%d%d",&n,&k); for(int i=0;i<n;i++) scanf("%d",&a[i]); memset(dp,0,sizeof(dp)); dp[a[0]%k][0]=1; int flag=0; for(int i=1;i<n;i++) { a[i]%=k; flag^=1; for(int j=0;j<k;j++) { if(dp[j][flag^1]) { dp[(j+a[i])%k][flag]=max(dp[(j+a[i])%k][flag^1],dp[j][flag^1]+1); } else dp[(j+a[i])%k][flag]=dp[(j+a[i])%k][flag^1]; } } printf("%d\n",max(dp[0][0],dp[0][1])); return 0; }