2017西安交大ACM小学期数据结构 [分块、二维矩阵]
Problem B
發(fā)布時(shí)間: 2017年6月28日 10:06?? 最后更新: 2017年6月28日 16:35?? 時(shí)間限制: 2000ms?? 內(nèi)存限制: 32M
描述給定一個(gè)n×m的矩形, 其中第i行第j列的值為ai,j
給出q個(gè)操作, 操作有兩種
對(duì)于形如1?x1?y1?x2?y2?z?的操作, 將(x1,y1)-(x2,y2)這段矩形區(qū)域的所有元素加上z, 滿(mǎn)足1≤x1≤x2≤n,?1≤y1≤y2≤m,?1≤z≤105
對(duì)于形如2?x?y的操作, 輸出ax,y, 滿(mǎn)足1≤x≤n,?1≤y≤m
998≤n≤1000,?998≤m≤1000,?9×104≤q≤105,?1≤ai,j≤105
輸入第一行三個(gè)整數(shù)n,?m,?q, 意義如上所述。
接下來(lái)n行, 每行m個(gè)整數(shù), 表示序列a。
再接下來(lái)q行, 每行第一個(gè)數(shù)為opt, 如果opt=1, 則后面緊跟五個(gè)數(shù), 意義如上所述; 如果opt=2, 則后面緊跟兩個(gè)數(shù), 意義如上所述。
對(duì)于所有操作2, 輸出答案, 一行一個(gè)。
樣例輸入1?復(fù)制 3 4 3 3 1 4 1 5 9 2 6 5 3 5 8 2 2 2 1 1 1 3 4 6 2 2 2 樣例輸出1 9 15題解:
這道題目給出的操作是對(duì)要做區(qū)間修改,端點(diǎn)查詢(xún)。我們可以仿照《數(shù)據(jù)結(jié)構(gòu)訓(xùn)練1》的思路,構(gòu)造出一個(gè)序列b,然后單點(diǎn)查詢(xún)變成了對(duì)序列b進(jìn)行求和,區(qū)間修改變成了
對(duì)序列b進(jìn)行的單點(diǎn)修改。
經(jīng)過(guò)嘗試,我們定義b[i][j] = a[i][j]+a[i-1][j-1]-a[i][j-1]-a[i-1][j]
這樣的話 sum(b[i][j])就等于a[i][j]
而我們對(duì)于區(qū)間x1,y1,x2,y2進(jìn)行修改(加上在z)就相當(dāng)于
在點(diǎn)b[x1][y1]處加上z
在b[x1][y2+1]處減去z
在b[x2+1][y1]處減去z
在b[x2+1][y2+1]處加上z
做以上4步操作。
這樣的話求和就很簡(jiǎn)單了,現(xiàn)在可以用兩種方法來(lái)實(shí)現(xiàn)求和
比較簡(jiǎn)單的一種想法是用二維樹(shù)狀數(shù)組來(lái)維護(hù),這里就不多說(shuō)了。
還有一種思路就是用分塊的方法,在O(q(sqrt n*m))的復(fù)雜度內(nèi)完成
由于二維矩陣在分塊上比較復(fù)雜。
我們轉(zhuǎn)向?qū)υ?xún)問(wèn)進(jìn)行分塊。
每塊地大小是s = sqrt(n*m)
每經(jīng)過(guò)s個(gè)詢(xún)問(wèn),我們都要對(duì)b數(shù)組和sum數(shù)組進(jìn)行重構(gòu)。
時(shí)間復(fù)雜度就是O(q(sqrt n*m))
代碼:
#include <cstdio> #include <cmath> const int MAX = 1007; typedef long long LL; LL cx1[100007]; LL cy1[100007]; LL cx2[100007]; LL cy2[100007]; LL cv[100007]; LL arr[MAX][MAX]; LL brr[MAX][MAX]; LL sum2[MAX][MAX]; LL tot = 0; LL las = 0; LL n,m,q; LL s; void cg(LL x1,LL y1,LL x2,LL y2,LL val){cx1[tot] = x1;cy1[tot] = y1;cx2[tot] = x2;cy2[tot] = y2;cv[tot] = val;tot ++;if(las + s == tot ){//重構(gòu) for(LL i = las;i < tot;i++){brr[cx1[i]][cy1[i]] += cv[i];brr[cx2[i]+1][cy2[i]+1] += cv[i];brr[cx1[i]][cy2[i]+1] -= cv[i];brr[cx2[i]+1][cy1[i]] -= cv[i];}las = tot;for(LL i = 1;i <= n;i++){for(LL j = 1;j <= m;j++){sum2[i][j] = sum2[i][j-1] + sum2[i-1][j] - sum2[i-1][j-1] + brr[i][j];}} } } LL sums(LL x,LL y){LL res = sum2[x][y];for(LL i = las;i < tot;i++){if(cx1[i] <= x && x <= cx2[i] && cy1[i] <= y && y <= cy2[i]){res += cv[i];}}return res; }int main(){//freopen("2.in","r",stdin);//freopen("2.out","w",stdout);scanf("%lld%lld%lld",&n,&m,&q);s = (LL)sqrt(n*m);for(LL i = 1;i <= n;i++){for(LL j = 1;j <= m;j++){scanf("%lld",&arr[i][j]);}}for(LL i = 1;i <= n;i++){for(LL j = 1;j <= m;j++){brr[i][j] = arr[i][j] + arr[i-1][j-1] - arr[i][j-1] - arr[i-1][j];}}for(LL i = 1;i <= n;i++){for(LL j = 1;j <= m;j++){sum2[i][j] = sum2[i][j-1] + sum2[i-1][j] - sum2[i-1][j-1] + brr[i][j];}} while(q--){LL opt;scanf("%lld",&opt);if(opt == 2){//查詢(xún) LL a,b;scanf("%lld%lld",&a,&b);printf("%lld\n",sums(a,b));sums(a,b);}else if(opt == 1){LL a,b,c,d,z;scanf("%lld%lld%lld%lld%lld",&a,&b,&c,&d,&z);cg(a,b,c,d,z);}}return 0;} /* 2 2 4 1 1 1 1 1 1 1 1 1 10 1 2 2 2 2 10 2 1 1 2 2 2 */
總結(jié)
以上是生活随笔為你收集整理的2017西安交大ACM小学期数据结构 [分块、二维矩阵]的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: iQOO Neo8 16+1TB顶配版今
- 下一篇: 2017西安交大ACM小学期数据结构 [