Codeforces 985G. Team Players

Description

有 \(n\) 个人 , \(m\) 对人有冲突 , 你要从这 \(n\) 个人中选出三个人成为一组 , 使得同一组的人不存在一对有冲突
题面

Solution

容斥
答案=总方案-至少有一个与 \(i\) 相连的+至少有 \(2\) 个与 \(i\) 相连的-\(i,j,k\) 都互相连接的方案数
前面三种都比较好求,讨论一下 \(i,j,k\) 的偏序关系就行了
关于第三种情况,需要 \(O(m^2)\) 枚举 , 只需要枚举度数较小的两个,最大的那个直接判断是否与前面两个相连就好了
这样做最坏复杂度是 \(O(n*\sqrt{n})\) 的 ,\(CF\) 的 \(Tutorial\) 有证明.

#include<bits/stdc++.h>
using namespace std;
template<class T>void gi(T &x){
    int f;char c;
    for(f=1,c=getchar();c<'0'||c>'9';c=getchar())if(c=='-')f=-1;
    for(x=0;c<='9'&&c>='0';c=getchar())x=x*10+(c&15);x*=f;
}
typedef unsigned long long ll;
const int N=2e5+10;
int n,m,head[N],nxt[N*4],to[N*4],num=0,Head[N],b[N],a[N];
ll s[N],ss[N],A,B,C,ans=0,in[N];
inline void link(int x,int y){nxt[++num]=head[x];to[num]=y;head[x]=num;}
inline void Link(int x,int y){nxt[++num]=Head[x];to[num]=y;Head[x]=num;}
int main(){
  freopen("pp.in","r",stdin);
  freopen("pp.out","w",stdout);
  int x,y;
  cin>>n>>m>>A>>B>>C;
  for(int i=1;i<=m;i++)gi(x),gi(y),link(x,y),link(y,x),in[x]++,in[y]++;
  for(int i=0,u;i<n;i++)
      for(int j=head[i];j;j=nxt[j])
          if(in[u=to[j]]>in[i]|| (in[u]==in[i] && u>i))Link(i,u);
  //+0                        
  for(int i=0;i<n;i++){
      s[i]=s[i-1]+i*A;
      ss[i]=ss[i-1]+s[i-1]+B*i*i;
      ans+=ss[i-1]+1ll*i*(i-1)/2*C*i;
  }
  for(int i=0;i<n;i++){
      m=0;
      for(int j=head[i];j;j=nxt[j])b[a[++m]=to[j]]=i+1;
      sort(a+1,a+m+1);s[m+1]=0;
      //-1
      for(int j=1;j<=m && a[j]<i;j++){
          ans-=(a[j]*A+i*B)*(n-i-1)+1ll*(i+n)*(n-i-1)/2*C;
          ans-=(a[j]*A+i*C)*(i-a[j]-1)+1ll*(a[j]+i)*(i-a[j]-1)/2*B;
          ans-=(a[j]*B+i*C)*a[j]+1ll*a[j]*(a[j]-1)/2*A;
      }
      //+2
      for(int j=1;j<=m && a[j]<i;j++){
          s[j]=s[j-1]+a[j];
          ans+=s[j-1]*A+(j-1)*(a[j]*B+C*i);
      }
      for(int j=m;j>=1 && a[j]>i;j--){
          s[j]=s[j+1]+a[j];
          ans+=s[j+1]*C+(m-j)*(a[j]*B+A*i);
      }
      for(int j=1;j<m;j++)
          if(a[j]<i && a[j+1]>i)ans+=s[j]*A*(m-j)+i*B*j*(m-j)+s[j+1]*C*j;
      //-3
      for(int j=Head[i];j;j=nxt[j])
          for(int k=Head[to[j]];k;k=nxt[k]){
              if(b[to[k]]!=i+1)continue;
              int c[3]={i,to[j],to[k]};
              sort(c,c+3);
              ans-=A*c[0]+B*c[1]+C*c[2];
          }  
  }
  cout<<ans<<endl;
  return 0;
}