题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3038
题目大意:给出n,m两个数,n代表一串数字序列中数字的个数(序列未知),m代表了m次询问,下面给出m行,每行有三个数字,A,B,S,代表从第A到第B个数的和为S,询问当前这一次S的值与上面的信息有无冲突,若有冲突则答案+1。
本题借鉴了这位大神的博客http://blog.csdn.net/niushuai666/article/details/6981689
但一开始被这位博客中的样例误导,错误样例如下,[1,5]的和为100,但[1,2]的和为200时冲突,但这是不对的。题目中只说数字为整数,但有可能为负数。所以只有当当前询问中的A与B在前几次询问中出现过后才有可能冲突,如[1,2]和为100,[1,5]的和为200,如给出[3,5]的和为300则与前两句冲突。
样例虽然错了但这位大神的思路却十分清晰。首先,我们将思路转化,A到B的和可以转化为,B减去A-1的差,本题的关键是设置一个数组val[i],这个数组中的值的意思是i与父节点的差。
(1)上文提到只有当当前询问中的A与B在前几次询问中出现过后才有可能冲突,所以,当A,B未出现时,求出A,B的父节点FA,FB,然后将FA,FB合并。合并原理如下,val[i]中存的是i与父节点的差,也可以理解为向量i—>Fi。
现在有两向量A—>FA , B—>FB ,需要合成FA—>FB,
由向量的性质可以轻易推出(FA—>FB) = (B—>FB) - (B—>FA)
又由上图可知 (B—>FA) = (A—>FA) - (A—>B)
所以我们最终可以推出(FA—>FB) = (B—>FB) - (A—>FA) + (A—>B)
即 ufs[fb]=fa;
val[fb]=val[a]-val[b]+s;
(2)第二种情况为给出的两个端点的父节点一致,则说明两者都已经出现过,这时仅需判断本次询问与我们已经储存的信息是否矛盾。
即判断 (A—>B) = (A—>FAB) - (B—>FAB)
即val[a]-val[b]=s
下面为代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cassert>
#include <ctime>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <functional>
#include <limits>
#include <vector>
#include <list>
#include <string>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <complex>
using namespace std;
#define debug cout<<"???"<<endl
#define sync std::ios::sync_with_stdio(false)
#define ll long long
#define INF 0x3f3f3f3f
#define frein freopen("in.txt", "r", stdin)
#define freout freopen("out.txt", "w", stdout)
const int MAXN=200010;
int n,m;
int ufs[MAXN];
int val[MAXN];
void init(int n)
{
for(int i=0;i<=n;i++)
ufs[i]=i,val[i]=0;
}
int finda(int x)
{
if(ufs[x]!=x)
{
int t=ufs[x];
ufs[x]=finda(ufs[x]);
val[x]+=val[t];
}
return ufs[x];
}
int main()
{
while(cin>>n>>m)
{
init(n);
int a,b,w;
int ans=0;
for(int i=1;i<=m;i++)
{
cin>>a>>b>>w;
a--;
int fa=finda(a);
int fb=finda(b);
if(fa!=fb)
{
ufs[fa]=fb;
val[fa]=val[b]-val[a]+w;
}
else
{
if(val[a]-val[b]!=w) ans++;
}
}
cout<<ans<<endl;
}
}