2021牛客暑期多校训练营3 B-Black and white（思维+最小生成树）

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;
}