埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛 L-K序列

埃森哲杯第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛 L-K序列

链接：https://www.nowcoder.com/acm/contest/91/L

来源：牛客网

题目描述

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

输入描述:

第一行为两个整数 n, K, 以空格分隔，第二行为 n 个整数，表示 a[1] ∼ a[n]，1 ≤ n ≤ $10^5$ , 1 ≤ a[i] ≤ $10^9$ , 1 ≤ nK ≤ $10^7$

输出描述:

输出一个整数表示最长子序列的长度 m

示例1

输入

7 5
10 3 4 2 2 9 8

输出

6

---

思路

这题固定姿势应该是用dp。

但我尝试了一下deque,双向队列（递归），发现也是可以滴，这里记一下。

具体思路是：

• 1）先假设最长子序列是序列长度，判断是否满足K序列。
• 2）若1）满足，则最长子序列长度是序列长度；若1）不满足，则分别减去队头队尾元素计算。
• 3）若2）满足，则最长子序列长度是序列长度-1；若2）不满足，则队头队尾出队，重复步骤 1）。

AC代码

#include <iostream>
#include <deque>
typedef long long ll;
using namespace std;
ll n,k;
ll ans;
deque<ll> que;
void rec(ll sum) {
    ll _front=que.front(),_back=que.back();
    if(sum%k==0){
        ans=que.size();
    } else if((sum-_front)%k==0||(sum-_back)%k==0){
        ans=que.size()-1;
    } else{
        que.pop_back();
        que.pop_front();
        rec(sum-_front-_back);
    }
}

int main() {
    cin >> n >> k;
    ll sum=0;
    for(ll i=0; i<n; i++) {
        ll a;
        cin >> a;
        que.push_back(a);
        sum+=a;
    }
    rec(sum);
    cout << ans << endl;
    return 0;
}