思维题——K序列

题目链接

思维题——K序列

题目描述

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

输入描述

第一行为两个整数 n, K, 以空格分隔，第二行为 n 个整数，表示 a[1] ∼ a[n]，1 ≤ n ≤ 10⁵ , 1 ≤ a[i] ≤ 10⁹ , 1 ≤ nK ≤ 10⁷。

输出描述

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

示例

输入

7 5
10 3 4 2 2 9 8

输出

6

方法一：

#include<bits/stdc++.h>
using namespace std;
int n,k;
long long a[100005],sum[100005];

int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		sum[i]=(sum[i-1]+a[i])%k;
	}
	int m=0;
	for(int i=0;i<=n;i++){
		for(int j=n;j>=i+m;j--){
			if(abs(sum[i]-sum[j])%k==0){
				m=max(m,abs(i-j));
			}
		}
	}
	cout<<m<<endl;
	return 0;
}

方法二：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int a[maxn],sum[maxn],n,k;
map<int,int> mp;

int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		sum[i]=(sum[i-1]+a[i])%k;
		if(!mp[sum[i]]){
			mp[sum[i]]=i;
		}
	}
	int ans=0;
	for(int i=1;i<=n;i++){
		ans=max(ans,i-mp[sum[i]]);
		if(sum[i]==0){
			ans=i;
		}
	}
	cout<<ans<<endl;
	return 0;
}