B-Black and white

效仿codeforces1012 B. Chemical table（并查集+思維），只不過邊帶權，是的n+m個點聯通，顯然就是求最小生成樹。

Code1 Kruskal

$O(n^2\log n^2)$

#include<bits/stdc++.h> using namespace std; using ll=long long; template <class T=int> T rd() {T res=0;T fg=1;char ch=getchar();while(!isdigit(ch)) {if(ch=='-') fg=-1;ch=getchar();}while( isdigit(ch)) res=(res<<1)+(res<<3)+(ch^48),ch=getchar();return res*fg; } const int N=400010; int fa[N]; int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);} int n,m,b,c,d,p,cur; int merge(int x,int y) {if(find(x)==find(y)) return 0;fa[find(x)]=find(y);--cur;return 1; } struct node {int u,v;int w;bool operator<(const node&o)const{return w<o.w;} }e[25000010]; int a[25000010]; int main() {n=rd(),m=rd(),a[0]=rd(),b=rd(),c=rd(),d=rd(),p=rd();int cnt=0;for(int i=1;i<=n;i++)for(int j=1;j<=m;j++){cnt++;a[cnt]=(1ll*a[cnt-1]*a[cnt-1]%p*b%p+a[cnt-1]*c%p+d)%p;e[cnt]={i,j+n,a[cnt]};}sort(e+1,e+1+cnt);cur=n+m;for(int i=1;i<=n+m;i++) fa[i]=i;ll ans=0;for(int i=1;i<=cnt;i++){if(cur==1) break;if(merge(e[i].u,e[i].v)) ans+=e[i].w;}printf("%lld\n",ans);return 0; }

Code2 prim

$O(n^2)$

#include<bits/stdc++.h> using namespace std; using ll=long long; template <class T=int> T rd() {T res=0;T fg=1;char ch=getchar();while(!isdigit(ch)) {if(ch=='-') fg=-1;ch=getchar();}while( isdigit(ch)) res=(res<<1)+(res<<3)+(ch^48),ch=getchar();return res*fg; } int n,m,A,B,C,D,P; int g[10010][10010]; int a[25000010]; int d[15010]; bool st[15010]; ll prim() {memset(d,0x3f,sizeof d);ll ans=0;d[1]=0;for(int i=0;i<n+m;i++){int t=-1;for(int j=1;j<=n+m;j++)if(!st[j]&&(t==-1||d[j]<d[t])) t=j;if(d[t]==0x3f3f3f3f) return d[t];st[t]=1;ans+=d[t];for(int j=1;j<=n+m;j++)d[j]=min(d[j],g[t][j]);}return ans; } int main() {n=rd(),m=rd(),A=rd(),B=rd(),C=rd(),D=rd(),P=rd();a[0]=A;memset(g,0x3f,sizeof g);for(int i=1;i<=n*m;i++){a[i]=(1ll*a[i-1]*a[i-1]%P*B%P+a[i-1]*C%P+D)%P;int u=(i-1)/m+1;int v=(i-1)%m+1;g[u][v+n]=g[v+n][u]=a[i];}printf("%lld\n",prim());return 0; }

總結

以上是生活随笔為你收集整理的2021牛客暑期多校训练营3 B-Black and white（思维+最小生成树）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。