Bug in Code CodeForces - 420C (计数,图论)

时间:2023-03-09 21:53:48
Bug in Code CodeForces - 420C (计数,图论)

大意: 给定$n$结点无向图, 共n条边, 有重边无自环, 求有多少点对(u,v), 满足经过u和v的边数>=p

可以用双指针先求出所有$deg_u+deg_v \ge p$的点对, 但这样会多算一些有公共边的

再枚举边, 减去 $deg_u+deg_v-cnt(u,v) < p$的即可

其中$deg$为点的度数, $cnt(u,v)$为$u$与$v$之间的边数

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <map>
#define REP(i,a,n) for(int i=a;i<=n;++i) using namespace std;
typedef long long ll;
typedef pair<int,int> pii; const int N = 2e6+10;
map<pair<int,int>,int> g;
int deg[N], n, p; int main() {
scanf("%d%d", &n, &p);
REP(i,1,n) {
int u, v;
scanf("%d%d", &u, &v);
if (u>v) swap(u,v);
++g[pii(u,v)],++deg[u],++deg[v];
}
ll ans = 0;
for (auto it:g) {
int x=it.first.first,y=it.first.second;
if (deg[x]+deg[y]>=p&&deg[x]+deg[y]-it.second<p) --ans;
}
sort(deg+1,deg+1+n);
int now = n;
REP(i,1,n) {
while (now>i&&deg[now]+deg[i]>=p) --now;
ans += n-max(i,now);
}
printf("%lld\n",ans);
}