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

最新推荐文章于 2021-08-24 20:11:58 发布

Bi08

最新推荐文章于 2021-08-24 20:11:58 发布

阅读量252

点赞数

分类专栏： 2021暑期训练

本文链接：https://blog.csdn.net/weixin_45765224/article/details/119062962

版权

2021暑期训练专栏收录该内容

32 篇文章 1 订阅

订阅专栏

题意

给你一个 $n * m$ 的矩阵，每个矩阵有一个权值，当一个正方形中三个点均已涂黑后，第四个点可免费涂黑，问将所有点涂黑所需的最小花费。

分析

所有点涂黑相当于将每行和每列均相连， $c [i] [j] = w$ 相当于从 $i$ 行走到 $j$ 列所花费权值为 $w$ ，建图跑 $p r i m$ 最小生成树即可。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 1e5 + 10;
int n, m, a, b, c, d, p;
vector<pii> G[N];
int f[2*N], cnt;
int find(int x){
    return f[x] == x ? f[x] : f[x] = find(f[x]);
}
int main(){
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin>>n>>m>>a>>b>>c>>d>>p;
    ll val = a;
    for(int i = 1; i <= n; i++)
       for(int j = 1; j <= m; j++){
           val = (val*val*b + val*c+d) % p;
           G[val].push_back({i, j+n});
       }
    for(int i = 0; i <= n + m; i++) f[i] = i;
    ll ans = 0;
    for(int w = 0; w < N; w++){
        for(auto i : G[w]){
           int u = find(i.first), v = find(i.second);
            if(u == v) continue;
            cnt++;
            f[u] = v;
            ans += w;
            if(cnt == n + m - 1){
                cout<<ans<<endl;
                return 0;
            }
        }
    }
}