HDU 3938

http://acm.hdu.edu.cn/showproblem.php?pid=3938

题意:
给你n个顶点,m条边,以及q次询问。
之后给出m条边的起末点及边权值,在q次询问,每次给你一个价值wal,问 这个价值你能在这个图里走多少条路。

思路:
1。我们可以看到查询量跟节点数量我们可以想到我们需要离线处理这道题,先把所有结果算出存储起来之后按顺序输出即可。
2。我们用两个结构体存储,一个存储图,一个存储查询信息,之后把查询信息的价值按从小到大排序,之后用kruskal 算法查询每个信息所得到的的答案。
3。之后把查询信息按标号sort一下,之后输出结果即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<stack>
#include<set>
using namespace std;
const int N= 10003;
const int M= 50002;
struct  Node
{
    int u,v;
    int len;
} edge[M];
bool cmp(Node a,Node b)
{
    return a.len<=b.len;
}
int fa[N];
int sum[N];
int n,m,q;
void init()
{
    for(int i=1; i<=n; i++)
        fa[i]=i,sum[i]=1;
}
int findx(int x)
{
    return fa[x]=fa[x]==x?x:findx(fa[x]);
}
int Union(int a,int b)
{
    int aa=findx(a);
    int bb=findx(b);
    if(aa==bb)
        return 0;
    else if(fa[aa]<fa[bb])
    {
        fa[bb]=aa;
        int tmp=sum[aa]*sum[bb];
        sum[aa]+=sum[bb];
        return tmp;
    }
    else
    {
        fa[aa]=bb;
        int tmp=sum[aa]*sum[bb];
        sum[bb]+=sum[aa];
        return tmp;
    }
}

int ans[N];
struct Q
{
    int L;
    int id;
    int ans;
} que[N];
bool cmp1(Q a,Q b)
{
    return a.L<=b.L;
}
bool cmp2(Q a,Q b)
{
    return a.id<b.id;
}
int main()
{
    while(scanf("%d%d%d",&n,&m,&q)!=EOF)
    {
        init();
        for(int i=0; i<m; i++)
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].len);
        for(int i=0; i<q; i++)
            {scanf("%d",&que[i].L);que[i].id=i;que[i].ans=0;}
        sort(edge,edge+m,cmp);
        sort(que,que+q,cmp1);
        int cnt=0;
        for(int i=0; i<q; i++)
        {
            while(edge[cnt].len<=que[i].L&&cnt<m)
            {
                //cout<<"FFF"<<endl;
                int aa=findx(edge[cnt].u);
                int bb=findx(edge[cnt].v);
                if(aa==bb)
                {
                    cnt++;
                    continue;
                }
                else
                {
                    que[i].ans+=Union(edge[cnt].u,edge[cnt].v);
                    cnt++;
                }
            }
            if(i>=1)
                que[i].ans+=que[i-1].ans;
        }
        sort(que,que+q,cmp2);
        for(int i=0; i<q; i++)
        {
           printf("%d\n",que[i].ans);
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值