一开始以为是动态规划,因为它其实是求满足特定要求的最大连续子列和及最长连续子列,所以就往动态规划求最大连续子列和的经典问题那个方向去思考,但是感觉行不通。
后来突然想到,可以把前i个数的和记录进sum[]数组里,每个sum[i]取余m会有一个值,如果sum[i]%m=0,那么他是满足条件的,如果sum[i]%m!=0,那么sum[i]应减去他前面与它取余m所得的数相同的sum[k],即sum[i]-sum[k],这样才可满足条件。
时间复杂度是O(N),可行。
代码如下:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
ll sum[maxn];
int p[maxn]; //p[i]记录下余数为1,2,3,...,m-1在sum数组中第一次出现的位置,注意是第一次哦
int main()
{
int n,m;
scanf("%d%d",&n,&m);
sum[0]=0;
memset(p,0,sizeof(p));
for(int i=1;i<=n;i++)
{
ll a;
scanf("%lld",&a);
sum[i]=sum[i-1]+a;
int temp=sum[i]%m;
if(!p[temp]) p[temp]=i;
}
ll ans1=0;
int ans2=0;
for(int i=1;i<=n;i++)
{
if(sum[i]%m==0)
{
ans1=max(ans1,sum[i]);
ans2=max(ans2,i);
}
else
{
int temp=sum[i]%m;
ans1=max(ans1,sum[i]-sum[p[temp]]);
ans2=max(ans2,i-p[temp]);
}
}
printf("%lld %d\n",ans1,ans2);
return 0;
}