【洛谷3396】哈希冲突(大力分块)

版权声明:本文为作者原创,若转载请注明地址。 https://blog.csdn.net/chenxiaoran666/article/details/82624184

点此看题面

大致题意:给你一个长度为 n 的数组 v a l 以及 m 个操作,操作有两种:一种是将 v a l x 修改为 y ,另一种操作是求出 v a l i ( i % x = y )


朴素的暴力

我们先来写一个朴素的暴力,代码如下:

int main()
{
    register int i,Q,x,y,ans;register char op;
    for(read(n),read(Q),i=1;i<=n;++i) read(a[i]);
    while(Q--)
    {
        read_alpha(op),read(x),read(y);
        if(op^'A') a[x]=y;//对于修改操作直接O(1)修改
        else
        {
            for(ans=0,i=y;i<=n;i+=x) ans+=a[i];//据题意统计答案
            write(ans),pc('\n');//输出
        }
    }
    return fwrite(Fout,1,FoutSize,stdout),0;
}

我们可以惊喜地发现,这样就有 91 分了。如果你写得足够优秀,说不定可以直接AC。


预处理答案+暴力更新

我们可以用 a n s i , j 来记录询问 i , j 时的答案。

首先, O ( n 2 ) 预处理出答案,并用 O ( n 2 ) 大小的数组把答案存下来,对于询问,可以 O ( 1 ) 输出答案,并 O ( n ) 暴力更新。

然后,我们悲剧地发现 n 150000 ,还没等到 T L E ,直接内存炸飞了。。。

代码略。


如何优化

我们可以对第二个上天了的代码进行优化。

记录全部答案毕竟所耗内存太大了,能不能只记录一部分的答案,然后另一部分暴力算呢?

就可以考虑分块

我们可以先 O ( n n ) 暴力预处理模数 n 时的答案,并用 a n s 数组将其存储下来。

然后,对于修改,我们可以在 O ( n ) 的时间复杂度内枚举每一个小于等于 n 的模数,然后更新对应的 a n s

对于询问,我们分两种情况讨论:

  • 如果模数 O ( n ) ,就直接输出 a n s
  • 如果模数 > O ( n ) ,就暴力求解,且枚举的数的个数肯定小于 O ( n )

这样的算法应该是 O ( ( n + m ) n ) 的。


一个玄学的小优化

虽然我个人认为最优块的大小应该是 O ( n ) ,但是,实践证明,以 O ( n 3 ) 为块的大小要比 O ( n ) 快很多。


代码

#include<bits/stdc++.h>
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
#define abs(x) ((x)<0?-(x):(x))
#define LL long long
#define ull unsigned long long
#define swap(x,y) (x^=y,y^=x,x^=y)
#define Fsize 100000
#define tc() (FinNow==FinEnd&&(FinEnd=(FinNow=Fin)+fread(Fin,1,Fsize,stdin),FinNow==FinEnd)?EOF:*FinNow++)
#define pc(ch) (FoutSize<Fsize?Fout[FoutSize++]=ch:(fwrite(Fout,1,Fsize,stdout),Fout[(FoutSize=0)++]=ch))
#define N 150000
#define SIZE 400 
int FoutSize=0,OutputTop=0;char Fin[Fsize],*FinNow=Fin,*FinEnd=Fin,Fout[Fsize],OutputStack[Fsize];
using namespace std;
int n,Q,blo,a[N+5],ans[SIZE][SIZE];
inline void read(int &x)
{
    x=0;static char ch;
    while(!isdigit(ch=tc()));
    while(x=(x<<3)+(x<<1)+ch-48,isdigit(ch=tc()));
}
inline void read_alpha(char &x)
{
    while(!isalpha(x=tc()));
} 
inline void write(int x)
{
    if(!x) return (void)pc('0');
    while(x) OutputStack[++OutputTop]=x%10+48,x/=10;
    while(OutputTop) pc(OutputStack[OutputTop]),--OutputTop;
}
int main()
{
    register int i,j,x,y,res;register char op;
    for(read(n),read(Q),blo=pow(n,0.33333),i=1;i<=n;++i) read(a[i]);//读入数据,这里的块的大小取n^(1/3)要比n^(1/2)快很多
    for(i=1;i<=n;++i) for(j=1;j<=blo;++j) ans[j][i%j]+=a[i];//O(n^(4/3))暴力预处理出模数≤n^(1/3)时的答案
    while(Q--)
    {
        read_alpha(op),read(x),read(y);
        if(op^'A')//对于修改操作
        {
            for(i=1;i<=blo;++i)//枚举每一个小于等于n^(1/3)的模数,修改对应答案
                ans[i][x%i]+=y-a[x];//减去原来的值,加上更新之后的值
            a[x]=y;//修改
        }
        else//对于询问操作
        {
            if(x<=blo) {write(ans[x][y]),pc('\n');continue;}//如果模数≤n^(1/3),直接输出预处理得到的答案
            for(i=y,res=0;i<=n;i+=x) res+=a[i];//否则,暴力统计答案
            write(res),pc('\n');//输出答案
        }
    }
    return fwrite(Fout,1,FoutSize,stdout),0;
}

猜你喜欢

转载自blog.csdn.net/chenxiaoran666/article/details/82624184