题目:数石子 rqnoj 36

题目描述

佳佳是个贪玩的孩子。一天，他正在跟凡凡玩“数石子”的游戏。佳佳在地上摆了Ｎ堆石子，其中第Ｉ堆石子有Ａi个石头。佳佳每次都会问凡凡：“凡凡，请问从第Ｉ堆到第Ｊ堆，总共有多少个石子？”聪明的凡凡每次都能快速而准确地回答对。凡凡老是被问问题，心里有些不服，就对佳佳说：“佳佳，你还记得你问了什么问题，我回答了什么答案吗？”佳佳说当然记得。于是凡凡说：“好，我把石子拿走，再问你一些相似的问题，你能答得出来吗？”佳佳张圆了嘴巴，望着凡凡，一脸疑问和惊讶的表情。你现在知道了游戏规则和过程，但没看见原来的石子。请你写一个程序来帮助佳佳。
数据范围
10%的数据满足1<=n<=10,0<=m,k<=10
30%的数据满足1<=n<=500,0<=m,k<=500
100%的数据满足1<=n<=5000,0<=m,k<=10000

输入格式

输入文件的第一行有３个数Ｎ(1<=n<=5000)，Ｍ，Ｋ(0<=m,k<=10000)，表示Ｎ堆石子，佳佳问了Ｍ个问题，凡凡要问Ｋ个问题。接下来Ｍ行每行３个整数Ｌ，Ｒ（1<=l<=r<=n），Ｘ(-108<=X<=108)，表示佳佳问从Ｌ堆到Ｒ堆的石子共有多少个，而凡凡回答Ｘ个。接下来Ｋ行每行２个整数Ａ，Ｂ(1<=a<=b<=n)，每行表示凡凡问从Ａ到Ｂ这些堆里的石子有多少个。

输出格式

输出文件需要对于凡凡提出每一个提问，你若可以回答，则输出答案，若无法回答，输出UNKNOWN。

并查集。

将 N 堆石子想象成 N 条连续的线段。

i-1~i 表示一条线段，所以 i-1~i 表示一堆石子，所以 l-1~r 表示 l 到 r 堆石子。

#include<iostream>
using namespace std;

int n,m,k,f[5005],d[5005]={0}; 

int Find(int node){
    if(node==f[node]) return node;
    int fn=f[node]; 
    f[node]=Find(f[node]);
    d[node]+=d[fn]; 
    return f[node]; 
    } 
    
int main()
{
    int i,x,y,l,fx,fy; 
    cin>>n>>m>>k;
    
    for(i=0;i<=n;++i)
    f[i]=i;
    
    for(i=1;i<=m;++i)
    {
      cin>>x>>y>>l;
      fx=Find(x-1),fy=Find(y); 
      if(fx==fy) continue; 
      f[fx]=fy;d[fx]=l+d[y]-d[x-1]; 
            } 
    for(i=1;i<=k;++i)
    {
      cin>>x>>y; 
      fx=Find(x-1),fy=Find(y);
      if(fx!=fy) cout<<"UNKNOWN"<<endl;
      else cout<<d[x-1]-d[y]<<endl; 
            } 
    return 0;
    }