題目描述
小明有一套玩具，一共包含NxM個(gè)部件。這些部件擺放在一個(gè)包含NxM個(gè)小格子的玩具盒中，每個(gè)小格子中恰好擺放一個(gè)部件。

每一個(gè)部件上標(biāo)記有一個(gè)0~9的整數(shù)，有可能有多個(gè)部件標(biāo)記相同的整數(shù)。

小明對(duì)玩具的擺放有特殊的要求：標(biāo)記相同整數(shù)的部件必須擺在一起，組成一個(gè)矩形形狀。

如以下擺放是滿足要求的：

00022
00033
44444

12244
12244
12233

01234
56789

以下擺放不滿足要求：

11122
11122
33311

111111
122221
122221
111111

11122
11113
33333

給出一種擺放方式，請(qǐng)你判斷是否符合小明的要求。
輸入
輸入包含多組數(shù)據(jù)。
第一行包含一個(gè)整數(shù)T，代表數(shù)據(jù)組數(shù)。 (1 <= T <= 10)
以下包含T組數(shù)據(jù)。
每組數(shù)據(jù)第一行包含兩個(gè)整數(shù)N和M。 (1 <= N, M <= 10)
以下包含N行M列的矩陣，代表擺放方式。
輸出
對(duì)于每組數(shù)據(jù)，輸出YES或者NO代表是否符合小明的要求。
樣例輸入
3
3 5
00022
00033
44444
3 5
11122
11122
33311
2 5
01234
56789
樣例輸出
YES
NO
YES
思路：我以為這個(gè)題挺難的呢，因?yàn)榭磀otcpp上面的通過(guò)率不高。但是數(shù)據(jù)量很小，就算是暴力的話，也應(yīng)該能過(guò)。我用了樹狀數(shù)組，但是時(shí)間復(fù)雜度還是挺高的。
我的辦法是：對(duì)于x，我們分別往右往下遍歷，找到這樣一個(gè)矩形，判斷這個(gè)矩形元素和。找元素和用樹狀數(shù)組節(jié)省時(shí)間。如果說(shuō)這個(gè)矩形中元素都是x，就符合條件。否則就不符合。如果當(dāng)前的這個(gè)元素，之前已經(jīng)處理過(guò)了，就說(shuō)明不符合條件。對(duì)于符合條件的，已經(jīng)處理過(guò)的元素，我們將它處理成-1，以免重復(fù)處理。這一過(guò)程我用的二重循環(huán)，在這里會(huì)浪費(fèi)時(shí)間，不知道有沒(méi)有什么好辦法。
代碼如下：

#include<bits/stdc++.h> #define ll long long using namespace std;const int maxx=15; int c[maxx][maxx]; int a[maxx][maxx]; int vis[maxx]; int n,m;inline int lowbit(int x){return x&-x;} inline void update(int x,int y,int v) {while(x<maxx){int ty=y;while(ty<maxx){c[x][ty]+=v;ty+=lowbit(ty);}x+=lowbit(x);} } inline int query(int x,int y) {int ans=0;while(x){int ty=y;while(ty){ans+=c[x][ty];ty-=lowbit(ty);}x-=lowbit(x);}return ans; } int main() {int t;scanf("%d",&t);while(t--){scanf("%d%d",&n,&m);for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) scanf("%1d",&a[i][j]);memset(c,0,sizeof(c));memset(vis,0,sizeof(vis));for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) update(i,j,a[i][j]);int flag=1;for(int i=1;i<=n;i++){for(int j=1;j<=m;j++){if(a[i][j]==-1) continue;if(vis[a[i][j]]) {flag=0;break;}int x=i,y=j;while(a[x][j]==a[i][j]&&x<=n) x++;while(a[i][y]==a[i][j]&&y<=m) y++;x-=1;y-=1;int sum=query(x,y)-query(x,j-1)-query(i-1,y)+query(i-1,j-1);if(sum!=a[i][j]*(x-i+1)*(y-j+1)) {flag=0;break;}vis[a[i][j]]=1;for(int kk=i;kk<=x;kk++) for(int zz=j;zz<=y;zz++) a[kk][zz]=-1;}if(!flag) break;}if(flag) cout<<"YES"<<endl;else cout<<"NO"<<endl;} }

ps：二維樹狀數(shù)組在處理這種二維矩陣的元素和上，真的挺有優(yōu)勢(shì)的。
努力加油a啊，(^o)/~

總結(jié)

以上是生活随笔為你收集整理的[蓝桥杯][2018年第九届真题]整理玩具（树状数组）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。