D Transitivity
题解:
根据题意可以推出结论: 如果存在连通块,那么这个连通块要满足条件,必然是满连通块.
一共有两种情况
1. 存在一个连通块不是满连通块
设cnt表示连通块的节点个数, num表示连通块边的个数
一个连通块的贡献 = cnt*(cnt-1)/2 - num;
那么最终答案 = 连通块贡献之和
2.所有连通块都是满连通块
因为我们至少需要添加一条边,所以此时等价于我们需要把两个连通块合并.
假设连通块A有x个节点,连通块B有y个节点,那么我们需要添加 x*y条边 才能满足条件.
所以即找到 最小和次小的连通块即可,答案 = x*y
AC代码:
#include <bits/stdc++.h>
typedef long long ll;
typedef unsigned long long ull;
#define int long long
#define endl '\n'
#define bit(x) (1ll << x)
using namespace std;
const int N = 1e6 + 5;
const int inf = 1e16;
vector<int> g[N];
int sz[N];//连通块大小
int cnt[N];//边的数量
int vis[N];
void solve()
{
int n,m;
cin >> n >> m;
for(int i = 1; i<=m; i++)
{
int u,v;
cin >> u >> v;
g[u].push_back(v);
g[v].push_back(u);
}
int Min1 = inf;//最小值
int Min2 = inf;//次小
auto dfs = [&](auto self, int u, int fa,int root)-> void
{
vis[u] = 1;
sz[u] = 1;
cnt[root]+=g[u].size();
for(auto v: g[u])
{
if(vis[v])
{
continue;
}
self(self,v,u,root);
sz[u]+=sz[v];
}
};
auto cal = [&](int now, int sum)//计算贡献
{
return sum*(sum-1)/2 - now;
};
int ans = 0;
int f = 0;
for(int i = 1; i<=n; i++)
{
if(vis[i])
{
continue;
}
dfs(dfs,i,-1,i);
cnt[i]/=2;
int val = cal(cnt[i],sz[i]);//连通块的贡献
if(val != 0)
{
ans+=val;
f = 1;
}
else
{
if(sz[i] < Min1)
{
Min2 = Min1;
Min1 = sz[i];