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

每一個部件上標記有一個0~9的整數，有可能有多個部件標記相同的整數。

小明對玩具的擺放有特殊的要求：標記相同整數的部件必須擺在一起，組成一個矩形形狀。

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

00022
00033
44444

12244
12244
12233

01234
56789

以下擺放不滿足要求：

11122
11122
33311

111111
122221
122221
111111

11122
11113
33333

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

#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：二維樹狀數組在處理這種二維矩陣的元素和上，真的挺有優勢的。
努力加油a啊，(^o)/~

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。