[最小生成树] 牛客第三场B-Black and white

题目

不想写了呜呜 这篇博客写的相当的好大家快来看这篇->博客

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define int long long
using namespace std;
const int INF = 0x3f3f3f3f;
int f[10010];
int find(int x){
	return x==f[x]?x:f[x]=find(f[x]);
}
vector<pair<int,int>> v[100010];

signed main(){
	int n,m,a,b,c,d,p;
	cin>>n>>m>>a>>b>>c>>d>>p;
	
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			a=(a*a*b+a*c+d)%p;
			v[a].push_back({i,n+j});
		}
	}
	for(int i=1;i<=n+m;i++) f[i]=i;
	bool flag=0;
	int ans=0,cnt=0;
	for(int i=0;i<p;i++){
		for(int j=0;j<v[i].size();j++){
			int x=v[i][j].first,y=v[i][j].second;
			int fx=find(x),fy=find(y);
			if(fx!=fy){
				f[fy]=fx;
				ans+=i;
				cnt++;
				if(cnt==n+m-1){
					flag=1;
					break;
				}
			}
		}
		if(flag) break;
	}
	cout<<ans<<endl;
    return 0;
}