wa自动机 的 莫队 刷题记录

洛谷P2709小B的询问

• 莫队裸题，模板题
• 莫队就是把询问区间排个序，先按左端点的Pos排序（POS是分块那个数组）,pos一样的按右端点排序
• 代码：

  #include <bits/stdc++.h>
  #define nmax 50010
  
  using namespace std;
  int a[nmax],ta[nmax],cnt[nmax]={0};
  int n,m,qn,k;
  struct que{
      int l,r,p,id;
      bool operator < (const que& x) const{ return (x.p==p)?(x.r>r):(x.p>p); }
  }q[nmax];
  
  int main(){
      scanf("%d%d%d",&n,&qn,&k);
      m=sqrt(n);
      for (int i=1; i<=n; i++) scanf("%d",&a[i]);
      for (int i=1; i<=qn; i++) {
          scanf("%d%d",&q[i].l,&q[i].r);
          q[i].p=(q[i].l-1)/m;
          q[i].id=i;
      }
      sort(q+1,q+qn+1);
      int l=q[1].l,r=q[1].l-1,ans=0;
      for (int i=1; i<=qn; i++) {
          while(l>q[i].l) l--,ans+=(2*cnt[a[l]]+1),cnt[a[l]]++;
          while(r<q[i].r) r++,ans+=(2*cnt[a[r]]+1),cnt[a[r]]++;
          while(l<q[i].l) ans-=(2*cnt[a[l]]-1),cnt[a[l]]--,l++;
          while(r>q[i].r) ans-=(2*cnt[a[r]]-1),cnt[a[r]]--,r--;
          ta[q[i].id]=ans;
      }
      for (int i=1; i<=qn; i++) printf("%d\n",ta[i]);
      return 0;
  }

 

BZOJ3289: Mato的文件管理

• 莫队加树状数组
• 每次区间长度增加的时候加上移动次数，减少的时候减去本来的移动次数（题意是只能移动相邻的嘛）
• 移动次数就是在当前区间的排名辣（因为要移动到那里去嘛，这个用树状数组很好维护。
• 没注意看题目，没有离散化re到死（果然人傻就是有各种各样的问题，嘤）
• 代码：

  #include <bits/stdc++.h>
  #define lowbit(x) x&(-x)
  #define nmax 50005
  
  using namespace std;
  typedef long long ll;
  int n,q,bn; //bn为每块的大小
  int data[nmax],c[nmax]={0},lsh[nmax];
  ll ans[nmax]={0};
  struct seg{
      int l,r,pl,id;
      bool operator < (const seg a) const{ return (a.pl==pl)?(a.r>r):(a.pl>pl); }
  }qes[nmax];
  
  void upd(int x,int y){ //y决定是加是减
      while(x<=n) { c[x]+=y; x+=lowbit(x); }
  }
  
  int myfind(int x){
      int ans=0;
      while(x>0) { ans+=c[x]; x-=lowbit(x); }
      return ans;
  }
  
  bool mycmp(int a,int b){ return data[a]<data[b]; }
  
  int main(){
      cin>>n;
      bn=sqrt(n);
      for (int i=1; i<=n; i++) { scanf("%d",&data[i]); lsh[i]=i; }
      sort(lsh+1,lsh+n+1,mycmp);
      for (int i=1; i<=n; i++) data[ lsh[i] ]=i;
      cin>>q;
      for (int i=1; i<=q; i++) {
          scanf("%d%d",&qes[i].l,&qes[i].r);
          qes[i].id=i;
          qes[i].pl=(qes[i].l-1)/bn+1;
      }
      sort(qes+1,qes+q+1);
      int l=qes[1].l,r=qes[1].l-1,nowl=0,ta=0;  //nowl现在该区间的长度
      for (int i=1; i<=q; i++) {
          while(qes[i].l<l) l--,upd(data[l],1),ta+=( myfind(data[l])-1 ),nowl++;
          while(qes[i].r>r) nowl++,r++,upd(data[r],1),ta+=( nowl-myfind(data[r]) );
          while(qes[i].l>l) ta-=( myfind(data[l])-1 ),upd(data[l],-1),nowl--,l++;
          while(qes[i].r<r) ta-=( nowl-myfind(data[r]) ),upd(data[r],-1),nowl--,r--;
          ans[ qes[i].id ]=ta;
      }
      for (int i=1; i<=q; i++) printf("%lld\n",ans[i]);
      return 0;
  }