Codeforces Round #920 (Div. 3) F. Sum of Progression（根号分治）

halllo__world__

已于 2024-03-01 23:12:15 修改

阅读量424

点赞数 9

分类专栏： 2024.2题解报告文章标签：算法

于 2024-02-14 22:46:08 首次发布

本文链接：https://blog.csdn.net/halllo__world__/article/details/136116443

版权

2024.2题解报告专栏收录该内容

12 篇文章 0 订阅

订阅专栏

文章介绍了一种使用分治策略解决给定数组的查询问题，当查询步长d大于数组长度的平方根时采用暴力求解，而当d小于根号n时进行后缀和预处理，通过计算后缀和数组快速得到答案。作者给出了C++代码示例。

摘要由CSDN通过智能技术生成

原题链接:F. Sum of Progression
题目大意：
给你一个由 n个数字组成的数组 a。还有 q个形式为 s,d,k的查询。
对于每个查询 q，求元素 a_s+a_s+d⋅2+⋯+a_{s+d⋅(k−1)⋅k}的和。即数组中从 s开始的 k个元素的和，步长为 d，乘以所得序列中元素的序号。
数据范围：1≤n≤10⁵,1≤q≤2⋅10⁵,−10⁸≤a₁,…,a_n≤10⁸，1≤s,d,k≤n, s+d⋅(k−1)≤n

解题思路：
若采用暴力做法，时间复杂度为O(t·q·k),会超时，考虑根号分治。
一.当d大于√n时,时间复杂度不会超过O(√n),可以直接暴力求解

int ans=0,l=1;
for(int i=s,l=1;l<=k;i+=d,l++)ans+=a[i]*l;

二.当d小于√n时,对数组作后缀和预处理，时间复杂度为O(n√n)

int s1[100405][400];//s1[i][j]表示以i为起点，步长为j的后缀和数组
int s2[100405][400];//s2表示以i为起点，步长为j的后缀和之和数组,即i为起点，步长为j一直加到末尾之和
//预处理
for(int i=n;i>=1;i--){
		for(int j=1;j<=p;j++){
			s1[i][j]=s1[i+j][j]+a[i]; 
		}
	}
	for(int i=n;i>=1;i--){
		for(int j=1;j<=p;j++){
			s2[i][j]=s2[i+j][j]+s1[i][j];
		}
	}

在这里插入图片描述
所求的答案即为三角形减去右上角和右下角的数

int ans=s2[s][d];
ans-=k*s1[s+d*k][d];//右上角
ans-=s2[s+d*k][d];//右下角

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
int a[100005];
int n;
int s1[100405][400];//s1[i][j]表示以i为起点，步长为j的后缀和数组
int s2[100405][400];//s2表示以i为起点，步长为j的后缀和之和数组,即i为起点，步长为j一直加到末尾之和
void solve(){
	int n,q;
	cin>>n>>q;
	int p=sqrt(n);
	for(int i=1;i<=n+p;i++){
		for(int j=1;j<=p;j++){
			s1[i][j]=0;
			s2[i][j]=0;
		}
	}
	for(int i=1;i<=n;i++)cin>>a[i];
	//预处理
	for(int i=n;i>=1;i--){
		for(int j=1;j<=p;j++){
			s1[i][j]=s1[i+j][j]+a[i]; 
		}
	}
	for(int i=n;i>=1;i--){
		for(int j=1;j<=p;j++){
			s2[i][j]=s2[i+j][j]+s1[i][j];
		}
	}
	while(q--){
		int s,d,k;
		cin>>s>>d>>k;
		//d>sqrt(n)时，暴力求解
		if(d>=p){
			int ans=0,l=1;
			for(int i=s,l=1;l<=k;i+=d,l++)ans+=a[i]*l;
			cout<<ans<<'\n';
		}
		//d<sqrt(n)时，减去右上角和右下角的数
		else {
			int ans=s2[s][d];
			ans-=k*s1[s+d*k][d];//右上角
			ans-=s2[s+d*k][d];//右下角
			cout<<ans<<'\n';
		}
	}
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int t=1;
	cin>>t;
	while(t--)solve(); 
    return 0;
 }