P1527 [国家集训队]矩阵乘法 整体二分 + 二维树状数组
生活随笔
收集整理的這篇文章主要介紹了
P1527 [国家集训队]矩阵乘法 整体二分 + 二维树状数组
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
傳送門
題意:
思路: 算是個整體二分的板子啦,不過這個是二維的矩陣,我們只需要把一位樹狀數組改成二維的,讓后動態維護單點加,區間查詢前綴和即可。
//#pragma GCC optimize(2) #include<cstdio> #include<iostream> #include<string> #include<cstring> #include<map> #include<cmath> #include<cctype> #include<vector> #include<set> #include<queue> #include<algorithm> #include<sstream> #include<ctime> #include<cstdlib> #define X first #define Y second #define L (u<<1) #define R (u<<1|1) #define pb push_back #define mk make_pair #define Mid (tr[u].l+tr[u].r>>1) #define Len(u) (tr[u].r-tr[u].l+1) #define random(a,b) ((a)+rand()%((b)-(a)+1)) #define db puts("---") #define lowbit(x) ((x)&(-x)) using namespace std;//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); } //void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); } //void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }typedef long long LL; typedef unsigned long long ULL; typedef pair<int,int> PII;const int N=1000010,mod=1e9+7,INF=0x3f3f3f3f; const double eps=1e-6;int n,m; int a[N],tot,ans[N]; int tr[510][510]; struct Node {int x1,y1,val,op,x2,y2,id; }q[N],q1[N],q2[N];void add(int x,int y,int c) {for(int i=x;i<=n;i+=lowbit(i))for(int j=y;j<=n;j+=lowbit(j))tr[i][j]+=c; }int sum(int x,int y) {int ans=0;for(int i=x;i;i-=lowbit(i))for(int j=y;j;j-=lowbit(j))ans+=tr[i][j];return ans; }int get_sub(int x1,int y1,int x2,int y2) {int ans=0;ans+=sum(x2,y2);ans+=sum(x1-1,y1-1);ans-=sum(x1-1,y2);ans-=sum(x2,y1-1);return ans; }void solve(int l,int r,int left,int right) {if(l>r||left>right) return;if(l==r){for(int i=left;i<=right;i++) if(q[i].op==2) ans[q[i].id]=a[l];return;}int mid=l+r>>1;int cnt1,cnt2; cnt1=cnt2=0;for(int i=left;i<=right;i++){if(q[i].op==1){if(q[i].val<=a[mid]) add(q[i].x1,q[i].y1,1),q1[++cnt1]=q[i];else q2[++cnt2]=q[i];}else{int s=get_sub(q[i].x1,q[i].y1,q[i].x2,q[i].y2);if(q[i].val<=s) q1[++cnt1]=q[i];//別寫錯符號!else q[i].val-=s,q2[++cnt2]=q[i];}}for(int i=1;i<=cnt1;i++) if(q1[i].op==1) add(q1[i].x1,q1[i].y1,-1);for(int i=1;i<=cnt1;i++) q[i+left-1]=q1[i];for(int i=1;i<=cnt2;i++) q[left+cnt1+i-1]=q2[i];solve(l,mid,left,left+cnt1-1); solve(mid+1,r,left+cnt1,right); }int main() { // ios::sync_with_stdio(false); // cin.tie(0);scanf("%d%d",&n,&m);for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) { int x; scanf("%d",&x); q[++tot]={i,j,x,1}; a[tot]=x; }sort(a+1,a+1+n*n);for(int i=1;i<=m;i++){int x1,y1,x2,y2,k; scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&k);q[++tot]={x1,y1,k,2,x2,y2,i};}solve(1,n*n,1,tot);for(int i=1;i<=m;i++) printf("%d\n",ans[i]);return 0; } /**/總結
以上是生活随笔為你收集整理的P1527 [国家集训队]矩阵乘法 整体二分 + 二维树状数组的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: P3403 跳楼机 同余最短路
- 下一篇: 大麦若叶的功效与作用、禁忌和食用方法