拼多多2021笔试真题集第三题

tt1724369779

于 2023-03-08 22:38:53 发布

阅读量153

点赞数

文章标签：面试职场和发展

本文链接：https://blog.csdn.net/tt1724369779/article/details/129413361

版权

多多的求和计算

多多路上从左到右有N棵树（编号1～N），其中第i个颗树有和谐值Ai。

多多鸡认为，如果一段连续的树，它们的和谐值之和可以被M整除，那么这个区间整体看起来就是和谐的。
现在多多鸡想请你帮忙计算一下，满足和谐条件的区间的数量。

题目地址：

牛客网 - 找工作神器|笔试题库|面试经验|实习招聘内推，求职就业一站解决_牛客网 (nowcoder.com)

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 256M，其他语言512M

输入：

第一行，有2个整数N和M，表示树的数量以及计算和谐值的参数。
（ 1 <= N <= 100,000, 1 <= M <= 100  ）
第二行，有N个整数Ai, 分别表示第i个颗树的和谐值。
（ 0 <= Ai <= 1,000,000,000 ）

输出：

共1行，每行1个整数，表示满足整体是和谐的区间的数量。

输入例子：

5 2
1 2 3 4 5

输出例子：

例子说明：

长度为1: [2], [4]
长度为2: 无
长度为3: [1,2,3], [3,4,5]
长度为4: [1,2,3,4], [2,3,4,5]
长度为5: 无
共6个区间的和谐值之和可以被2整除。

#include <iostream>
#include <string.h>
using namespace std;

int main() {
    int n,m;
    cin>>n>>m;
    long long  arr[100100];
    for(int i=0;i<n;i++){
        scanf("%lld",&arr[i]);
        arr[i]%=m;
    }
    for(int i=1;i<n;i++){
        arr[i]=(arr[i]+arr[i-1])%m;
    }
    long long  sum[200];
    memset(sum,0,sizeof(sum));
    for(int i=0;i<n;i++){
        sum[arr[i]]++;
    }
    long long ans=0;
    for(int i=0;i<m;i++){
        long long temp=0;
        if(sum[i]%2==0){
            temp=sum[i]/2;
            temp*=(sum[i]-1);
        }
        else{
            temp=(sum[i]-1)/2;
            temp*=sum[i];
        }
        ans+=temp;
    }
    ans+=sum[0];
    cout<<ans<<endl;
    return 0;
}
// 64 位输出请用 printf("%lld")