蓝桥杯 k倍区间

题目描述在这里不过多阐述了，在这里附上题目链接https://www.lanqiao.cn/problems/97/learning/?page=1&first_category_id=1&sort=students_count&second_category_id=3&difficulty=20

下面直接开始介绍算法思路：

首先，阅读完题目我立刻就想到了动态规划的做法，时间复杂度是O（n2），下面附上代码

#include <iostream>
#include <vector>
using namespace std;
int main()
{
  int n,k;
  cin>>n>>k;
  vector<int> a(n, 0);
  vector<long long> dps(n,1);
  for(int i=0;i<n;++i){
    cin>>a[i];
    dps[i] = a[i];
  }
  int ans = 0;
  dps[0]=a[0];
  if(dps[0]%k==0)++ans;
  for(int i=1;i<n;++i){
    dps[i]=a[i]+dps[i-1];
    if(dps[i]%k==0)++ans;
  }
  for(int i=1;i<n;++i){
    for(int j=i;j<n;++j){
      if((dps[j]-dps[i-1])%k==0)++ans;
    }
  }
  cout<<ans<<endl;
  return 0;
}

结果不出所料，果然超时了，通过了两个用例

接着我绞尽脑汁都想不出更好的解法，所以我阅读了其他人的解法，以下附上代码：时间复杂度为O（n）

#include<iostream>

using namespace std;

const int N =100010;
int n,k;
long long s[N],cnt[N];
long long res;

int main()
{
  cnt[0] = 1;
  cin >> n >> k;

  for(int i = 1; i <= n; i ++ )
  {
      cin >> s[i];
      s[i] += s[i - 1];

      res += cnt[s[i] % k];
      cnt[s[i] % k]++ ;
  }

  cout << res;
  return 0;
}

但是我始终无法理解这三行代码：

s[i] += s[i - 1];
res += cnt[s[i] % k];
cnt[s[i] % k]++ ;

直到有一刻我幡然醒悟，这种感觉很爽，下面是我记录的想法：

首先分析第一行代码，其目的是为了求出前i个数之和（即s[i]代表前i个数之和）

其次分析第二行代码，在说第二代码之前，先说一说cnt数组是什么，cnt[i]存储的是遍历到i个输入时，s[i]中出现余数为i的个数。那么为什么要加上这个呢？首先，除了cnt[0]初始化为1外，其余都初始化为0，当再有出现相同余数时就说明这期间加的数之和恰好为k（很妙，就像时钟一样走了一轮），那么最后一个与之前出现过的余数为i可以组成的组合为多少呢，就是cnt[i]，所以res += cnt[s[i] % k]。至于为什么cnt[0]初始化为1呢，这很好理解，是因为当余数为0时就代表这个数单个就可以组成一个区间，不需要再等待下一个相同的余数（就是不需要再来一轮）

最后cnt[s[i] % k]++就很好理解了，就是把出现的余数++