摘桃子（双指针）

最新推荐文章于 2024-07-21 13:38:21 发布

to cling

最新推荐文章于 2024-07-21 13:38:21 发布

阅读量147

点赞数 1

分类专栏：双指针文章标签：动态规划区间和整数k map 代码实现

本文链接：https://blog.csdn.net/QQ2530063577/article/details/125130969

版权

双指针专栏收录该内容

4 篇文章 0 订阅

订阅专栏

代码源 div1 201

Solution

$sum_r - sum_l) \% k = r - l$
$sum_r - r) \% k = (sum_l - l) \% k$
注意：r - l < k

Code

const int N = 2e5 + 5;
ll a[N], sum[N];
map<ll, int> mp;

int main()
{
	IOS;
	int n, k; cin >> n >> k;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
		sum[i] = sum[i - 1] + a[i];
	}

	mp[0]++;
	ll ans = 0, l = 0;
	for (int i = 1; i <= n; i++)
	{
		if (i - l >= k)//保证r - l合法
		{
			mp[(sum[l] - l + k) % k]--;
			l++;
		}
		ans += mp[(sum[i] - i + k) % k];
		mp[(sum[i] - i + k) % k]++;
		
	}
	cout << ans << endl;


	return 0;
}