CF 1093E Intersection of Permutations——CDQ分治

题目:http://codeforces.com/contest/1093/problem/E

只能想到转化成查询一个区间里值在一个范围里的数的个数……

没有想到这样适合用主席树套树状数组维护。不过据说卡空间。

参考了这里的题解:https://www.luogu.org/problemnew/solution/CF1093E

写了CDQ分治。一个值在两个序列里的位置看成两维坐标的话,就是查平面区域内点的个数。

按 x (在第一个序列里的位置)排序,分治操作的时间,y (在第二个序列里的位置)用树状数组解决。

做这一层的时候要先枚举,再消除影响,然后再把 a[ ] 分成两部分。这样才能保证贡献给这个询问的是 x 在它之前、时间在它之前的那些 y 。在树状数组里就不考虑 x 了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e5+5;
int n,m,ta[N],tb[N],pa[N],pb[N],ans[N],tot,f[N],cnt;
struct Node
{
  bool fx;int x,y,k,id,tim;
  Node(bool f=0,int x=0,int y=0,int k=0,int i=0,int t=0):
    fx(f),x(x),y(y),k(k),id(i),tim(t) {}
  bool operator< (const Node &b)const
  {return x<b.x||(x==b.x&&tim<b.tim);}
}a[(N<<2)+N],b[(N<<2)+N];//+N
int rdn()
{
  int ret=0;bool fx=1;char ch=getchar();
  while(ch>'9'||ch<'0'){if(ch=='-')fx=0;ch=getchar();}
  while(ch>='0'&&ch<='9')ret=ret*10+ch-'0',ch=getchar();
  return fx?ret:-ret;
}
void add(int x,int k){for(;x<=n;x+=(x&-x))f[x]+=k;}
int query(int x){int ret=0;for(;x;x-=(x&-x))ret+=f[x];return ret;}
void solve(int l,int r)
{
  if(l>=r)return;
  int mid=l+r>>1,p0=l,p1=mid+1;
  for(int i=l;i<=r;i++)
    if(!a[i].fx&&a[i].tim<=mid)add(a[i].y,a[i].k);
    else if(a[i].fx&&a[i].tim>mid)ans[a[i].id]+=a[i].k*query(a[i].y);
  for(int i=l;i<=r;i++)
    if(!a[i].fx&&a[i].tim<=mid)add(a[i].y,-a[i].k);
  for(int i=l;i<=r;i++)
    if(a[i].tim<=mid)b[p0++]=a[i];
    else b[p1++]=a[i];
  for(int i=l;i<=r;i++)a[i]=b[i];
  solve(l,mid); solve(mid+1,r);
}
int main()
{
  n=rdn();m=rdn();
  for(int i=1;i<=n;i++)ta[i]=rdn(),pa[ta[i]]=i;
  for(int i=1;i<=n;i++)tb[i]=rdn(),pb[tb[i]]=i;
  for(int i=1;i<=n;i++)
    a[++cnt]=Node(0,pa[i],pb[i],1,0,cnt);//1
  for(int i=1,op,x1,y1,x2,y2,w1,w2;i<=m;i++)
    {
      op=rdn();
      if(op==1)
    {
      tot++;
      x1=rdn();x2=rdn();y1=rdn();y2=rdn();
      x1--; y1--;
      a[++cnt]=Node(1,x2,y2,1,tot,cnt);
      a[++cnt]=Node(1,x1,y2,-1,tot,cnt);
      a[++cnt]=Node(1,x2,y1,-1,tot,cnt);
      a[++cnt]=Node(1,x1,y1,1,tot,cnt);
    }
      else
    {
      int w1=rdn(),w2=rdn();
      y1=w1;x1=pa[tb[w1]];y2=w2;x2=pa[tb[w2]];
      swap(pb[tb[w1]],pb[tb[w1]]);
      swap(tb[w1],tb[w2]);
      a[++cnt]=Node(0,x1,y1,-1,0,cnt);
      a[++cnt]=Node(0,x2,y2,-1,0,cnt);
      a[++cnt]=Node(0,x1,y2,1,0,cnt);
      a[++cnt]=Node(0,x2,y1,1,0,cnt);
    }
    }
  sort(a+1,a+cnt+1);
  solve(1,cnt);
  for(int i=1;i<=tot;i++)printf("%d\n",ans[i]);
  return 0;
}

猜你喜欢

转载自www.cnblogs.com/Narh/p/10147087.html