hihocoder 1515 分数调查(树形dp)

hihocoder 1515 分数调查

时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
小Hi的学校总共有N名学生,编号1-N。学校刚刚进行了一场全校的古诗文水平测验。

学校没有公布测验的成绩,所以小Hi只能得到一些小道消息,例如X号同学的分数比Y号同学的分数高S分。

小Hi想知道利用这些消息,能不能判断出某两位同学之间的分数高低?

输入
第一行包含三个整数N, M和Q。N表示学生总数,M表示小Hi知道消息的总数,Q表示小Hi想询问的数量。

以下M行每行三个整数,X, Y和S。表示X号同学的分数比Y号同学的分数高S分。

以下Q行每行两个整数,X和Y。表示小Hi想知道X号同学的分数比Y号同学的分数高几分。

对于50%的数据,1 <= N, M, Q <= 1000

对于100%的数据,1 <= N, M, Q<= 100000 1 <= X, Y <= N -1000 <= S <= 1000

数据保证没有矛盾。

输出
对于每个询问,如果不能判断出X比Y高几分输出-1。否则输出X比Y高的分数。

样例输入
10 5 3
1 2 10
2 3 10
4 5 -10
5 6 -10
2 5 10
1 10
1 5
3 5
样例输出
-1
20
0

可以将这些人与他们的关系看成一个图,一张图中是存在很多独立的分量的,我们只要对于每个分量随意选择一个点作为根,然后用树形dp求出其他点到这个点的距离即可,那么成绩即为两个点到根的距离之差。当然此题使用加权并查集也是可做的。

#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#define ll long long
#define inf 1000000000
#define mod 1000000007
using namespace std;
int read()
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
const int Max=100000+10;
const int E=(100000+10)*2;
int head[Max],nex[E],pnt[E],cost[E],edge;
int vex[Max<<1],R[Max<<1],vis[Max],dis[Max],first[Max],tot;
int n;
void Addedge(int u,int v,int c)
{
    pnt[edge]=v;
    cost[edge]=c;
    nex[edge]=head[u];
    head[u]=edge++;
}
void dfs(int u,int deep)
{
    vis[u]=1;
    vex[++tot]=u;
    first[u]=tot;
    R[tot]=deep;
    for(int x=head[u]; x!=-1; x=nex[x])
    {
        int v=pnt[x],c=cost[x];
        if(!vis[v])
        {
            dis[v]=dis[u]+c;
            dfs(v,deep+1);
            vex[++tot]=u;
            R[tot]=deep;
        }
    }
}
void Init()
{
    edge=0;
    memset(head,-1,sizeof(head));
    memset(nex,-1,sizeof(nex));
    memset(vis,0,sizeof(vis));
}
int book[Max],fa[Max];
int find(int x){
    return fa[x]==-1?x:fa[x]=find(fa[x]);
}
int main()
{
    Init();
    int n=read(),m=read(),q=read();
    int u,v,c,f1,f2;
    memset(fa,-1,sizeof(fa));
    for(int i=0; i<m; i++)
    {
        u=read();
        v=read();
        c=read();
        f1=find(u);f2=find(v);
        if(f1!=f2) fa[f2]=f1;
        Addedge(u,v,c);
        Addedge(v,u,-c);
        book[u]=book[v]=1;
    }
    tot=0;
    for(int i=n;i>=1;i--) if(book[i]){
       if(dis[i]!=0) continue;
       dis[i]=0;
       dfs(i,i);
    }
    int ans;
    while(q--)
    {
        u=read();v=read();
        ans=dis[v]-dis[u];
        if(find(u)!=find(v)) ans=-1;
        printf("%d
",ans);
    }
    return 0;
}

原文地址:https://www.cnblogs.com/zsyacm666666/p/6806002.html