2021年北京理工大学ACM CLUB清明节组队训练赛 G Generator Grid

题目

思路很简单，玩玩两个样例大概就知道可以贪心+并查集了，一开始写复杂了，其实代码简单的一匹。

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<sstream>
#include<queue>
#include<list>
#include<bitset>
#include<stack>
#include<bitset>
#include<iomanip> 
#include<set>
#include<cmath>
#include<map>
#include<functional>
#define int long long
#define IO ios::sync_with_stdio(false)
#define _BYTE unsigned char
#define HIBYTE(x) (*((_BYTE*)&(x)+1))
#define eps 1e-8
#define double long double
using namespace std;
struct node
{
	int x,y,val;
}a[300005];
bool cmp(node x,node y)
{
	return x.val<y.val;
}
int n,m,fa[300005],cnt;
int fin(int x)
{
	if(fa[x]==x)return x;
	return fa[x]=fin(fa[x]);
}
signed main()
{
	IO;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		cnt++;
		cin>>a[cnt].y>>a[cnt].val;
	}
	for(int i=1;i<n;i++)
	{
		cnt++;
		a[cnt].x=i,a[cnt].y=i+1;
		cin>>a[cnt].val;
	}
	for(int i=1;i<=n;i++)fa[i]=i;
	cnt++;
	int vall;cin>>vall;
	a[cnt].x=n,a[cnt].y=1,a[cnt].val=vall;
	sort(a+1,a+cnt+1,cmp);
	int ans=0;
	for(int i=1;i<=cnt;i++)
	{
		int x=fin(a[i].x),y=fin(a[i].y);
		if(x==y)continue;
		ans+=a[i].val;
		fa[x]=y;
	}
	cout<<ans;
}