刁姹接到一个任务,为税务部门调查一位商人的账本,看看账本是不是伪造的。账本上记录了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
题意:给你n个月份,m条记录,每条记录记录从s到t的账单,根据这几条记录判断这个账单是不是对的。
分析:带权并查集。 比普通并查集只是多了一个权值,记录更新下权值就行了。
找root节点代码:
int getf(int x)
{
if(fa[x]==x)
return fa[x];
int temp=getf(fa[x]);
sum[x]+=sum[fa[x]]; //更新自己到root节点的权值
return fa[x]=temp;
}
判断是否具有相同的根节点:
若有相同根节点判断月份的金钱是不是匹配。
若不相同,靠左原则,把右节点的根节点加到左节点的根节点并更新sum(可以利用向量相等sum[x]+z=sum[y]+sum[fy])。
代码:
void merge(int x,int y,int z)
{
int fx,fy;
fx=getf(x);
fy=getf(y);
if(fx!=fy)
{
fa[fy]=fx;
sum[fy]=sum[x]-sum[y]+z;
}
else
{
if(sum[y]-sum[x]!=z)
flag=1;
}
}
最后注意一点 从s到t月份的金钱不是sum[t]-sum[s](这是算得s+1到t的金钱),应该是sum【t】-sum【s-1】 (前缀和思想)。
AC代码:
#include<iostream>
using namespace std;
int fa[1010],sum[1010],n,m,flag;
void init(int n)
{
for(int i=0;i<=n;i++)
{
fa[i]=i;
sum[i]=0;
}
}
int getf(int x)
{
if(fa[x]==x)
return fa[x];
int temp=getf(fa[x]);
sum[x]+=sum[fa[x]];
return fa[x]=temp;
}
void merge(int x,int y,int z)
{
int fx,fy;
fx=getf(x);
fy=getf(y);
if(fx!=fy)
{
fa[fy]=fx;
sum[fy]=sum[x]-sum[y]+z;
}
else
{
if(sum[y]-sum[x]!=z)
flag=1;
}
}
int main()
{
int t,x,y,z;
cin>>t;
while(t--)
{
flag=0;
cin>>n>>m;
init(n);
for(int i=1;i<=m;i++)
{
cin>>x>>y>>z;
x--;
merge(x,y,z);
}
if(flag)
cout<<"false"<<endl;
else
cout<<"true"<<endl;
}
return 0;
}