Portal HDU - 3938 (并查集 +离线处理)

看了还久都没什么思路 看了题解思路还是不晓得咋写
最后看了别人的代码 才明白这题咋弄

题目意思

题目给定了点,边和边的边权 求最多有多少对点之间的路径上的最大值小于L

题目思路

这题要用到最小生成树kruscal的思想
我们要让点之间的路径最大值尽量小
直接对边的边权值排序 从小往大取
用并查集连接不连通的区域 并且记录区域点的数量
因为我们从小到大取 取得时候只要保证边权小于限制的值
就不需要考虑路径最大值超过限制
因为询问较多
如果在线操作复杂度会过高
所以离线操作会更好
记录以后以限制的大小排序 最后输出前再按读入顺序排序回去就好了

ac代码

 #include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include  <map>
#include <set>
#include <utility>
#define pi 3.1415926535898
#define ll long long
#define lson rt<<1
#define rson rt<<1|1
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define print1 printf("111\n")
using namespace std;
const int maxn = 1e6+5;
const int inf = 0x3f3f3f3f;
const ll llinf = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1000000007;
//998244353

int sum[maxn],f[maxn];
struct node
{
    int x,y,v;
}e[maxn];

struct point
{
    int id,l;
    int ans;
}ans[maxn];

bool cmp1(node a,node b)
{
    return a.v<b.v;
}

bool cmp2(point a,point b)
{
    return a.l<b.l;
}

bool cmp3(point a,point b)//用于按顺序输出
{
    return a.id<b.id;
}

int findx(int x)
{
    return x==f[x]?f[x]:f[x]=findx(f[x]);
}
int main()
{
    int n,m,q;
    while(scanf("%d%d%d",&n,&m,&q)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            sum[i]=1;
            f[i]=i;
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].v);
        }
        sort(e+1,e+1+m,cmp1);
        for(int i=1;i<=q;i++)
        {
            scanf("%d",&ans[i].l);
            ans[i].id=i;
            ans[i].ans=0;
        }
        sort(ans+1,ans+1+q,cmp2);
        int cnt=1;
        ans[0].ans=0;
        for(int i=1;i<=q;i++)
        {
            while(e[cnt].v<=ans[i].l&&cnt<=m)
            {
                int r1=findx(e[cnt].x);
                int r2=findx(e[cnt].y);
                if(r1!=r2)
                {
                    ans[i].ans+=sum[r1]*sum[r2];
                    f[r2]=r1;
                    sum[r1]+=sum[r2];
                }
                cnt++;
            }
            ans[i].ans+=ans[i-1].ans;
        }
        sort(ans+1,ans+1+q,cmp3);
        for(int i=1;i<=q;i++)
            printf("%d\n",ans[i].ans);
    }

}

你可能感兴趣的:(Portal HDU - 3938 (并查集 +离线处理))