第十六屆上海大學程式設計聯賽春季賽暨上海高校金馬五校賽:L-K序列(DP)
阿新 • • 發佈:2019-02-15
題意:給一個數組 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; }