BZOJ1202:狡猾的商人(帶權並查集)
1202: [HNOI2005]狡猾的商人
題目鏈接:https://www.lydsy.com/JudgeOnline/problem.php?id=1202
Description:
刁姹接到一個任務,為稅務部門調查一位商人的賬本,看看賬本是不是偽造的。賬本上記錄了n個月以來的收入情況,其中第i 個月的收入額為Ai(i=1,2,3...n-1,n), 。當 Ai大於0時表示這個月盈利Ai 元,當 Ai小於0時表示這個月虧損Ai 元。所謂一段時間內的總收入,就是這段時間內每個月的收入額的總和。 刁姹的任務是秘密進行的,為了調查商人的賬本,她只好跑到商人那裏打工。她趁商人不在時去偷看賬本,可是她無法將賬本偷出來,每次偷看賬本時她都只能看某段時間內賬本上記錄的收入情況,並且她只能記住這段時間內的總收入。 現在,刁姹總共偷看了m次賬本,當然也就記住了m段時間內的總收入,你的任務是根據記住的這些信息來判斷賬本是不是假的。
Input:
第一行為一個正整數w,其中w < 100,表示有w組數據,即w個賬本,需要你判斷。每組數據的第一行為兩個正整數n和m,其中n < 100,m < 1000,分別表示對應的賬本記錄了多少個月的收入情況以及偷看了多少次賬本。接下來的m行表示刁姹偷看m次賬本後記住的m條信息,每條信息占一行,有三個整數s,t和v,表示從第s個月到第t個月(包含第t個月)的總收入為v,這裏假設s總是小於等於t。
Output:
包含w行,每行是true或false,其中第i行為true當且僅當第i組數據,即第i個賬本不是假的;第i行為false當且僅當第i組數據,即第i個賬本是假的。
Sample Input:
2
3 3
1 2 10
1 3 -5
3 3 -15
5 3
1 5 100
3 5 50
1 2 51
Sample Output:
true
false
題解:
給出m段區間的區間和,問是否有矛盾。
題解:
我們通過分析,只有當給出區間類似於 [x,y] 以及 [z,y] 時,我們可以多求出一段[x,z)這段區間的和。
由於涉及到點的重合,我們可以考慮用加權並查集。設sum[x]是x到其父節點的區間和,即 (x,f[x] ]。一開始時,我們什麽都不知道sum即為0。
後來在查找和合並的時候,要對原來的並查集加以修改。合並時的修改思路可以用向量法去理解,查找時則是從父節點不斷更新值到當前結點,這麽想就好理解一些。
另外註意這裏我把輸入的左閉右閉區間修改成了左閉右開區間,發現這麽便於理解一些,代碼寫起來BUG沒那麽多。
代碼如下:
#include <cstdio> #include <cstring> #include <algorithm> #include <iostream> #include <cmath> using namespace std; const int N = 105 ; int f[N],sum[N]; int t,n,m; void init(){ memset(sum,0,sizeof(sum)); for(int i=0;i<=n+1;i++) f[i]=i; } int find(int x){ if(f[x]==x) return x; int tmp = f[x]; f[x]=find(f[x]); sum[x]+=sum[tmp]; return f[x]; } int main(){ scanf("%d",&t); while(t--){ scanf("%d%d",&n,&m); init(); bool flag =true ; for(int i=1,x,y,w;i<=m;i++){ scanf("%d%d%d",&x,&y,&w);y++; int fx = find(x),fy=find(y); if(fx==fy){ if(sum[y]-sum[x]!=w) flag = false ; }else{ f[fy]=fx; sum[fy]=sum[x]-sum[y]+w; } } if(flag) puts("true"); else puts("false"); } return 0; }
BZOJ1202:狡猾的商人(帶權並查集)