题目内容:
平面上有n个点(n≤8000),每两个点之间都有一条红色或者是黑色的线段,任意三点均不共线。
现在,已知哪些点之间连的线段是红色的,剩下的线段都是黑色的,要求计算这些点组成的三角
形中有多少是同色的(顶点编号从1到n)?
输入描述
第一行是n, m(3≤n≤8000),n表示点的个数,m表示红色线段的条数。下面m行,每
一行都是两个整数a和b,表示点a和点b之间的线段是红色的(a<b)。
输出描述
只有一行,表示同色三角形的个数。
提示:本题输出数据可能会超出长整数(long int)的范围。
输入样例
6 5
1 2
1 3
2 3
2 5
3 6
输出样例
7
算同色三角形并不好算, 但算异色三角形好算很多, 当组成异色三角形时, 必定有两个顶点的相邻两边为异色, 故可以算出所有异色三角形的个数, 用总个数减去其一半即可
#include<iostream>
#include<string>
#include<string.h>
#include<vector>
#include<cstdio>
#include<stack>
#include<algorithm>
using namespace std;
typedef long long ll;
vector<int> v[8003];
int n, m;
int main()
{
ios::sync_with_stdio(false);
cin >> n >> m;
int a, b;
for (int i = 0; i < m; i++)
{
cin >> a >> b;
v[a].push_back(b);
v[b].push_back(a);
}
ll sum = 0;
for (int i = 1; i <= n; i++)
sum += (v[i].size() * (n - v[i].size() - 1));
cout << (ll)n * (n - 1) * (n - 2) / 6 - sum /2 << endl;
}