首页 > 代码库 > hdu5737
hdu5737
首先思考一个朴素的做法
将b[]维护成一个线段树套有序表,每次修改a[]用线段树+lazy tag
并在线段树的子区间上在有序表中二分更新这段区间中a[i]>=b[i]的值,复杂度O(nlog^2)
有没有更优的做法?
考虑在一次修改操作中,查询的数是相同的,并且b[]这个有序表始终不变
因此在生成b[]的线段树套有序表过程中(其实就是归并排序的记录)
我们维护每个数在左右子区间中名次。
这样修改的时候我们只要对b[]排好序的整体做一次二分,找到b[m]<=x<b[m+1]
把修改成x当作修改成b[m],这样我们就能O(1)地更新每个子区间的计数,问题得解
1 #include<bits/stdc++.h> 2 #define mp make_pair 3 using namespace std; 4 const int mo=1000000007; 5 vector< pair<int,int> > tr[100010*4]; 6 int laz[100010*4],s[100010*4],a[100010],b[100010],c[100010]; 7 int A,B,n,m,t; 8 int C = ~(1<<31), M = (1<<16)-1; 9 int rnd(int last,int &a,int &b) 10 { 11 a = (36969 + (last >> 3)) * (a & M) + (a >> 16); 12 b = (18000 + (last >> 3)) * (b & M) + (b >> 16); 13 return (C & ((a << 16) + b)) % 1000000000; 14 } 15 16 void build(int i,int l,int r) 17 { 18 laz[i]=-2; 19 tr[i].clear(); 20 if (l==r) 21 { 22 s[i]=(a[l]>=b[l]); 23 } 24 else { 25 int m=(l+r)>>1; 26 build(i*2,l,m); 27 build(i*2+1,m+1,r); 28 s[i]=s[i*2]+s[i*2+1]; 29 int x=l,y=m+1,h=l; 30 while (x<=m&&y<=r) 31 { 32 if (b[x]<b[y]) 33 { 34 tr[i].push_back(mp(x-l,y-m-2)); 35 c[h++]=b[x++]; 36 } 37 else { 38 tr[i].push_back(mp(x-l-1,y-m-1)); 39 c[h++]=b[y++]; 40 } 41 } 42 while (x<=m) 43 { 44 tr[i].push_back(mp(x-l,r-m-1)); 45 c[h++]=b[x++]; 46 } 47 while (y<=r) 48 { 49 tr[i].push_back(mp(m-l,y-m-1)); 50 c[h++]=b[y++]; 51 } 52 for (int k=l; k<=r; k++) b[k]=c[k]; 53 } 54 } 55 56 void push(int i) 57 { 58 int x=laz[i]; 59 laz[i*2]=(x==-1)?-1:tr[i][x].first; 60 laz[i*2+1]=(x==-1)?-1:tr[i][x].second; 61 s[i*2]=laz[i*2]+1; 62 s[i*2+1]=laz[i*2+1]+1; 63 laz[i]=-2; 64 } 65 66 void add(int i,int l,int r,int x,int y,int w) 67 { 68 if (x<=l&&y>=r) 69 { 70 laz[i]=w; 71 s[i]=w+1; 72 } 73 else { 74 int m=(l+r)>>1; 75 if (laz[i]>-2) push(i); 76 if (x<=m) add(i*2,l,m,x,y,w==-1?w:tr[i][w].first); 77 if (y>m) add(i*2+1,m+1,r,x,y,w==-1?w:tr[i][w].second); 78 s[i]=s[i*2]+s[i*2+1]; 79 } 80 } 81 82 int ask(int i,int l,int r,int x,int y) 83 { 84 if (x<=l&&y>=r) return s[i]; 85 else { 86 int m=(l+r)>>1,ans=0; 87 if (laz[i]>-2) push(i); 88 if (x<=m) ans+=ask(i*2,l,m,x,y); 89 if (y>m) ans+=ask(i*2+1,m+1,r,x,y); 90 return ans; 91 } 92 } 93 94 int main() 95 { 96 int cas; 97 scanf("%d",&cas); 98 while (cas--) 99 { 100 scanf("%d%d%d%d",&n,&m,&A,&B); 101 for (int i=1; i<=n; i++) scanf("%d",&a[i]); 102 for (int i=1; i<=n; i++) scanf("%d",&b[i]); 103 b[n+1]=2147483647; 104 build(1,1,n); 105 int ans=0; 106 int last=0; 107 for (int i=1; i<=m; i++) 108 { 109 int l=rnd(last,A,B)%n+1,r=rnd(last,A,B)%n+1,x=rnd(last,A,B)+1; 110 if (l>r) swap(l,r); 111 if ((l+r+x)%2==0) 112 { 113 last=ask(1,1,n,l,r); 114 ans=(ans+1ll*i*last%mo)%mo; 115 } 116 else { 117 int w=upper_bound(b+1,b+1+n,x)-b-2; 118 add(1,1,n,l,r,w); 119 } 120 } 121 printf("%d\n",ans); 122 } 123 }
hdu5737
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。