洛谷3396 哈希冲突（分块）（预处理）（暴力）

题意

有n个数，分别为a[i]。对于询问操作，给定一个模数x，求所有下标mod x等于y的数的和。对于修改操作，把位置x上的数改成y。

 

思路

朴素做法不难想到，复杂度O（n^2）。

接着可以考虑预处理的问题了，毕竟询问有15W个。
设ans[p][k]表示模数是p，余数是k的数的和。
虽然询问O（1），但预处理O（n^2），还是不能接受。何况空间复杂度还是O（n^2）的。

先解决空间问题。能不能某些ans不提前处理呢？
我们想到，如果p越大，那么求和的数就会越少。为了均衡，我们把这个点定在了sqrt(n)上。这样1~sqrt(n)的询问可以O(1)回答；对于sqrt(n)+1~n的问题，我们采用暴力来求解，其时间复杂度也不过为O（sqrt(n)）。
考虑修改操作，也可以在O（sqrt(n)）时间内解决。

总的时间复杂度为O（(n+m)*sqrt(n)），在洛谷的高性能+O2下跑过了。
据一些大佬说，如果把sqrt(n)改成pow(n,0.33)会快得起飞，我也不明白这个玄学加速。代码还是要这样写的

 

题解

（分块思想：）预处理+暴力
以pow(n,0.33)为界，0~pow(n,0.33)预处理，pow(n,0.33)+1~n暴力求解。有修改操作暴力修改预处理的部分。

 

小结

一道难题不一定要单纯的用一种算法来解决，可以联合几种算法来解决（暴力也是一种算法），这对分块的思想要求很高。

 

代码

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxl=150010,maxsqr=70;

int n,m,maxp;
int value[maxl];
int ans[maxsqr][maxsqr];

int main()
{
    scanf("%d%d",&n,&m);
    maxp=pow(n,0.33);
    for(int i=1;i<=n;i++) scanf("%d",&value[i]);
    
    for(int p=1;p<=maxp;p++)
        for(int i=1;i<=n;i++) ans[p][i%p]+=value[i];
    
    int x,y;
    char opt[5];
    while(m--)
    {
        scanf("%s",opt);
        scanf("%d%d",&x,&y);
        if(opt[0]=='A')
        {
            if(x<=maxp) printf("%d\n",ans[x][y]);
            else
            {
                int sum=0;
                for(int i=y;i<=n;i+=x) sum+=value[i];
                printf("%d\n",sum);
            }
        }
        else
        {
            int c=y-value[x];
            value[x]=y;
            for(int p=1;p<=maxp;p++) ans[p][x%p]+=c;
        }
    }
    return 0;
}