2024 年第六届码蹄杯全国大学生程序设计大赛本科组决赛 2 4 7题-CSDN博客

本文链接：https://blog.csdn.net/xiaolemyzq/article/details/140519496

MC0369魔法徽章第7题

算法：组合数学

思路：考虑三种情况

第一种情况是一个中心点连着三个点，先枚举点，枚举点的连接点的个数，组合数学C（点的个数,3）从连接个数中选3个有多少种选法。

第二种情况是三个点互相连接的三角形，先枚举边，统计左右两个端点相同的连接点.

第三种情况是一条直线，三个点串起来，枚举中间的边，将中点的边的两边点所连接不同的点的个数统计一下，相同的也统计一下，左边连接的不同的点-1 乘上右边连接的不同的点-1（需要把他们两个点互相连接的一条边减去），再加上左边连接相同的点乘上右边连接的不同的点，和再加上左边连接不同的点乘上右边连接的相同的点即可。

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const ll Q = 2e5+9;
const ll MOD = 1e9 + 7;
pair<ll,ll> a[Q];
set<ll> st[Q];
ll qzh[Q];
ll C(ll n,ll m){
    return n*(n-1)*(n-2)/6;
}
void slove(){ 
    ll n,m;cin>>n>>m;
    for (ll i = 1; i <= m; i++)
    {
      cin>>a[i].first>>a[i].second;
      st[a[i].first].insert(a[i].second);
      st[a[i].second].insert(a[i].first);
    }
    ll sum=0;
    for (ll i = 1; i <= n; i++)
    {
      if(st[i].size()>=3)
      sum+=C((st[i].size()),3);
      sum%=MOD;
    }
    ll num=0;
    for (ll i = 1; i <= m; i++)
    {
      //cout<<sum<<"\n";
      ll o=a[i].first;
      ll p=a[i].second;
      map<ll,ll> mp;
      for(auto i:st[o]){
        mp[i]++;
      }
      for(auto i:st[p]){
        mp[i]++;
      }
      ll l=1,r=1;
      for(auto i:mp){
        if(i.second>=2) num++,l++,r++;
      }
      sum+=(st[o].size()-l)*(st[p].size()-r)%MOD;
      sum+=(st[o].size()-l)*(r-1);
      sum+=(st[p].size()-r)*(l-1);
      sum%=MOD;
      
    }
    sum+=num/3;
    cout<<sum%MOD;
}
int main(){
  ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
  ll _=1;//cin>>_;
  while(_--)slove();
  return 0;
}