蓝桥杯 历届试题 国王的烦恼(c++)

国王的烦恼

时间限制：1.0s 内存限制：256.0MB

问题描述
　　C国由n个小岛组成，为了方便小岛之间联络，C国在小岛间建立了m座大桥，每座大桥连接两座小岛。两个小岛间可能存在多座桥连接。然而，由于海水冲刷，有一些大桥面临着不能使用的危险。

如果两个小岛间的所有大桥都不能使用，则这两座小岛就不能直接到达了。然而，只要这两座小岛的居民能通过其他的桥或者其他的小岛互相到达，他们就会安然无事。但是，如果前一天两个小岛之间还有方法可以到达，后一天却不能到达了，居民们就会一起抗议。

现在C国的国王已经知道了每座桥能使用的天数，超过这个天数就不能使用了。现在他想知道居民们会有多少天进行抗议。
输入格式
　　输入的第一行包含两个整数n, m，分别表示小岛的个数和桥的数量。
　　接下来m行，每行三个整数a, b, t，分别表示该座桥连接a号和b号两个小岛，能使用t天。小岛的编号从1开始递增。
输出格式
　　输出一个整数，表示居民们会抗议的天数。
样例输入
4 4
1 2 2
1 3 2
2 3 1
3 4 3
样例输出
2
样例说明
　　第一天后2和3之间的桥不能使用，不影响。
　　第二天后1和2之间，以及1和3之间的桥不能使用，居民们会抗议。
　　第三天后3和4之间的桥不能使用，居民们会抗议。
数据规模和约定
　　对于30%的数据，1<=n<=20，1<=m<=100；
　　对于50%的数据，1<=n<=500，1<=m<=10000；
　　对于100%的数据，1<=n<=10000，1<=m<=100000，1<=a, b<=n， 1<=t<=100000。

思路：并查集、排序(按天数降序)


思路参考：https://blog.csdn.net/qiqi_skystar/article/details/50923574
并查集参考：https://blog.csdn.net/qq_41593380/article/details/81146850

超时代码(寻找根节点的代码未优化导致超时)

#include <iostream>
#include <algorithm>
 
using namespace std;

int f[10002];

struct un
{
	int x,y,d;
};

bool cmp(un a, un b)
{
	return a.d>b.d;
}

int find(int value)
{
	if(value==f[value])
		return value;
	else return find(f[value]);
}
 
int research(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y)
	{
		f[x]=y;
		return 1;
	}
	else return 0;
} 
 
int main()
{
    int n,m,i;
    cin>>n>>m;
    un s[m];
    for(i=1;i<=n;i++)
    	f[i]=i;
    for(i=0;i<m;i++)
    	cin>>s[i].x>>s[i].y>>s[i].d;
    sort(s,s+m,cmp);
    int ans,pre;
    ans=0;
    pre=-1;
    for(i=0;i<m;i++)
    {
    	if(research(s[i].x,s[i].y)==1&&s[i].d!=pre)
    	{
    		ans++;
    		pre=s[i].d;
    	}
    }
    cout<<ans;
    return 0;
}

正确代码(优化后)

#include <iostream>
#include <algorithm>
 
using namespace std;

int f[10002];

struct un
{
	int x,y,d;
};

bool cmp(un a, un b)
{
	return a.d>b.d;
}

int find(int value)
{
	int temp1,temp2;
	temp1=value;
	while(value!=f[value])
	{
		value=f[value];
	}
	while(temp1!=value)
	{
		temp2=f[temp1];
		f[temp1]=value;
		temp1=temp2;
	}
	return value;
}
 
int research(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y)
	{
		f[x]=y;
		return 1;
	}
	else return 0;
} 
 
int main()
{
    int n,m,i;
    cin>>n>>m;
    un s[m];
    for(i=1;i<=n;i++)
    	f[i]=i;
    for(i=0;i<m;i++)
    	cin>>s[i].x>>s[i].y>>s[i].d;
    sort(s,s+m,cmp);
    int ans,pre;
    ans=0;
    pre=-1;
    for(i=0;i<m;i++)
    {
    	if(research(s[i].x,s[i].y)==1&&s[i].d!=pre)
    	{
    		ans++;
    		pre=s[i].d;
    	}
    }
    cout<<ans;
    return 0;
}