洛谷 P3948 数据结构


传送门:洛谷 P3948 数据结构
题目描述:

维护一个数组,使其能区间修改,并动态查询某区间内满足\(min\leq(a_i\times i\ \%\ mod)\leq max\) 的数的个数


算法分析:该题操作共两个阶段——
1、在线:要求动态修改,但数据量很小(\(\leq 1000\)),可以暴力查询,区间修改,单点查询
2、离线:不要求动态修改,但数据量很大(\(\leq 10000000\)),可以预处理,然后区间查询
如上,跑两个树状数组即可。
事实证明,树状数组只能得72分,为了用数据结构(正解不用树状结构),吸个氧就过了(最大点490ms)


#include<iostream>
#include<cstdio>
#include<cstring>
#define node int
#define maxN 80000
#define in(x) x=read()
using namespace std;
class binary_tree
{
    public:
        inline node query(int x,int y) {return (y+1)*sum(tree1,y)-sum(tree2,y)-x*sum(tree1,x-1)+sum(tree2,x-1);}
        inline void build(node a[]) {for(register int i=1;i<=tot;i++) modify(i,a[i]-a[i-1]);}
        inline void update(node x,node y,node k) {modify(x,k); modify(y+1,-k);}
        inline void clear(int n) {tot=n; memset(tree1,0,sizeof(tree1)); memset(tree2,0,sizeof(tree2));}
    private:
        node tree1[maxN+1],tree2[maxN+1],tot;
        inline node lowbit(node x) {return x&(-x);}
        inline node sum(node a[],node x) {node ans=0; while(x>0) {ans+=a[x]; x-=lowbit(x);} return ans;}
        inline void modify(node x,node k) {for(register node i=x;i<=tot;i+=lowbit(i)) {tree1[i]+=k; tree2[i]+=k*x;}}
};
typedef int ll;
ll l,r,k,read(); char s[2];
int n,m,D,minT,maxT,check(int);
binary_tree treeA,treeB;
inline void task1(),task2();
int main()
{
    task1(); task2();
    return 0;
}
inline int check(int x)
{
    ll now=treeA.query(x,x)%D;
    if(minT<=((x*now)%D) && ((x*now)%D)<=maxT) return 1;
    return 0;
}
inline void task1()
{
    in(n); in(m); in(D); in(minT); in(maxT);
    treeA.clear(n);
    for(register int i=1;i<=m;i++)
    {
        scanf("%s",s); in(l); in(r);
        if(s[0]=='A') treeA.update(l,r,read());
        else
        {
            ll ans=0;
            for(register int j=l;j<=r;j++) ans+=check(j);
            printf("%d\n",ans);
        }
    }
}
inline void task2()
{
    in(m); treeB.clear(n);
    for(register int i=1;i<=n;i++) if(check(i)) treeB.update(i,i,1);
    for(register int i=1;i<=m;i++)
    {
        in(l); in(r);
        printf("%d\n",treeB.query(l,r));
    }
}
inline ll read()
{
    ll num=0,f=1;
    char ch=getchar();
    while((ch<'0' || ch>'9') && ch!='-') ch=getchar();
    if(ch=='-') {f=-1; ch=getchar();}
    while(ch>='0' && ch<='9')
    {
        num=(num<<3)+(num<<1)+ch-'0';
        ch=getchar();
    }
    return num*f;
}
关注微信公众号

注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



 
粤ICP备14056181号  © 2014-2020 ITdaan.com