1. 程式人生 > >第十六屆上海大學程式設計聯賽春季賽暨上海高校金馬五校賽:L-K序列(DP)

第十六屆上海大學程式設計聯賽春季賽暨上海高校金馬五校賽:L-K序列(DP)

題意給一個數組 a,長度為 n,若某個子序列中的和為 K 的倍數,那麼這個序列被稱為“K 序列”。現在要你 對陣列 a 求出最長的子序列的長度,滿足這個序列是 K 序列。

題解:看程式碼。

#include <bits/stdc++.h>
using namespace std;

const double EPS = 1e-8;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int maxn = 1e6 + 10;
int n, k;
int a[maxn];
int dp[maxn], tmp[maxn];

int main()
{
    scanf("%d%d", &n, &k);

    for(int i = 1; i <= n; i++){
        scanf("%d", &a[i]);
        for(int j = 0; j < k; j++) tmp[j] = dp[j];
        dp[a[i] % k] = max(1, dp[a[i] % k]);
        for(int j = 0; j < k; j++){
            if(tmp[j]) dp[(j + a[i]) % k] = max(dp[(j+a[i])%k], tmp[j]+1);
        }
    }

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

    return 0;
}