P8649 [蓝桥杯 2017 省 B] k 倍区间(前缀和+优化(桶分类))

分析:

(1)任意连续子序列可用两个前缀和的差来表示

(2)判断该子序列是否为k的倍数        p1-p2 模 0 (mod k)

等价于:前缀和模 k 是否同余

(3)同余的任意两前缀和组合的序列均满足k的倍数

(4)余数为0的需要特判,因为余数为0的前缀和本身也满足条件

实现:

1.利用滚动窗口,逐个计算前缀和模k的结果,

2.用桶数组统计模k对应不同余数的数量 

#include<iostream>
using namespace std;

long long count1[1000009] = {0};

int main()
{
	int n, k; cin >> n >> k;
	long long ans = 0,sum=0;
	for (int i = 0; i < n; i++)
	{
		long long temp = 0;
		scanf("%lld", &temp);
		sum = (sum + temp) % k;
		count1[sum]++;
	}
	// 单一前缀和为k的倍数(特判)
	ans += (count1[0] * (count1[0] + 1) / 2);

	for (int i = 1; i < k; i++)
		// 单一前缀和不为k的倍数
		ans += (count1[i] * (count1[i] - 1) / 2);
	

	cout << ans;
	return 0;
}

  • 18
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值