Vijos 1090题:连续数之和

来源:互联网 发布:汤臣倍健褪黑素片 知乎 编辑:程序博客网 时间:2024/06/05 18:01

描述

有n个正整数排成一行。你的目的是要从中取出一个或连续的若干个数,使它们的和能够被k整除。
例如,有6个正整数,它们依次为1、2、6、3、7、4。若k=3,则你可以取出1、2、6,或者2、6、3、7,也可以仅仅取出一个6或者3使你所取的数之和能被3整除。当然,满足要求的取法不止以上这4种。事实上,一共有7种取法满足要求。
给定n和k,以及这n个数。你的任务就是确定,从这n个数中取出其中一个数或者若干连续的数使它们的和能被k整除有多少方法。
由于取法可能很多,因此你只需要输出它mod 1234567的值即可。

格式

输入格式

第一行有两个正整数,分别代表n和k。输入数据保证有n<=500 000,k<=100 000。
以下n行每行一个正整数。这些正整数保证都不大于10 000。

输出格式

一个正整数。它应该是你的答案mod 1234567的结果。

样例1

样例输入1

6 3
1
2
6
3
7
4

样例输出1

7

限制

各个测试点1s


#include<iostream>using namespace std;int a[100001];int main(){    std::ios::sync_with_stdio(false);    int n, k, ans = 0, x, y = 0;    cin >> n >> k;    for (int i = 0;i < n;i++)    {        cin >> x;        y = (x + y) % k;        ans += a[y];        if (y == 0)ans++;        ans %= 1234567;        a[y]++;    }    cout << ans << endl;    return 0;}
1 0