hdu3938（并查集+离线）

最新推荐文章于 2025-08-14 22:56:31 发布

原创最新推荐文章于 2025-08-14 22:56:31 发布 · 1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

acm

本文详细解析了如何使用最小生成树算法解决特定问题，并通过两次代码优化展现了算法理解和实现过程中的关键步骤。从理解题意到算法应用，再到代码层面的具体实现与优化，为读者提供了一个完整的案例。

这道题的第一个难点在于题目，对于我等英语菜B来说，理解这个题意还真是有点难......

理解了题意之后，可以发现，用最小生成树的思路理解，如果可以加入某一条边使两个集合联通，那么增加的路径个数就是num【a】×num【b】。

所以我第一次写的代码如下：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#define N 10005
using namespace std;
int G[N],num[N];
struct len
{
    int a,b,leng;
}Len[5*N];
int findx(int x)
{
    return G[x]==x?x:G[x]=findx(G[x]);
}
bool cmp(len a,len b)
{
    return a.leng<b.leng;
}
int main()
{
    int n,m,q,ans;
    while(scanf("%d %d %d",&n,&m,&q)!=EOF)
    {
        int i,answer,j;
        for(i=0;i<m;i++)
        {
            scanf("%d %d %d",&Len[i].a,&Len[i].b,&Len[i].leng);
        }
        sort(Len,Len+m,cmp);
        for(i=0;i<q;i++)
        {
            scanf("%d",&ans);
            j=0,answer=0;
            for(j=0;j<=n;j++) G[j]=j,num[j]=1;
            j=0;
            while(Len[j].leng<=ans && j!=m)
            {
                int p=findx(Len[j].a),
                    q=findx(Len[j].b);
                if(p!=q)
                {
                    G[p]=q;
                    answer+=(num[q]*num[p]);
                    num[q]+=num[p];
                }
                j++;
            }
            printf("%d\n",answer);
        }
    }
    return 0;
}

提交了之后发现TLE了，经查看样例发现，有一部分的数据重复计算，增加了时间的损耗，所以我可以先记录，排序，后计算，由此：计算时间就变成了最大的那个数的计算时间，提交了AC

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#define N 10005
using namespace std;
int G[N],num[N],ans1[N];
struct len
{
    int a,b,leng;
}Len[5*N];
struct ans
{
    int a,i;
}answer[N];
int findx(int x)
{
    return G[x]==x?x:G[x]=findx(G[x]);
}
bool cmp(len a,len b)
{
    return a.leng<b.leng;
}
bool cmp2(ans a,ans b)
{
    return a.a<b.a;
}
int main()
{
    int n,m,q,ans;
    while(scanf("%d %d %d",&n,&m,&q)!=EOF)
    {
        int i,answer1,j;
        for(i=0;i<m;i++)
        {
            scanf("%d %d %d",&Len[i].a,&Len[i].b,&Len[i].leng);
        }
        for(i=0;i<q;i++)
        {
            scanf("%d",&answer[i].a);
            answer[i].i=i;
        }
        sort(Len,Len+m,cmp);
        sort(answer,answer+q,cmp2);
        for(j=0;j<=n;j++) G[j]=j,num[j]=1;
        j=0,answer1=0;
        for(i=0;i<q;i++)
        {
            while(Len[j].leng<=answer[i].a && j!=m)
            {
                int p=findx(Len[j].a),
                    q=findx(Len[j].b);
                if(p!=q)
                {
                    G[p]=q;
                    answer1+=(num[q]*num[p]);
                    num[q]+=num[p];
                }
                j++;
            }
            ans1[answer[i].i]=answer1;
        }
        for(i=0;i<q;i++)
        {
            printf("%d\n",ans1[i]);
        }
    }
    return 0;
}