MT3004·找四边形

题目:

样例输入 

4 12
1 2
1 3
1 4
2 1
2 3
2 4
3 1
3 2
3 4
4 1
4 2
4 3

样例输出

12

数据范围 

n,m(1 \leq n \leq 3000, 0 \leq m \leq 3000),a_{i},b_{i}(1 \leq a_{i},b_{i} \leq n;a_{i}\neq b_{i})

 算法设计

涉及的算法

        枚举和图论基础

        采用邻接矩阵g[N]来存储图,其中vector<ll> g[N]是建立了一个二维的vector

        来用sum记录每个点 i 到达点 j 共有多少条路径,然后每算一个点的就存储到数组sum中,最后计算出当前点 i 的结果,加到ans上,继续遍历下一个点

        其中可以利用公式C_{n}^{2} = \frac{n*(n-1)}{2}来计算

AC代码:

#include<bits/stdc++.h> 
#define ll long long
using namespace std;
#define N 3010
vector<ll> g[N];
ll n,m,sum[N];
ll C(ll x){
	return x*(x-1)/2;
}
int main( )
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>n>>m;
    for(ll i=1;i<=m;++i){
    	ll x,y;
    	cin>>x>>y;
    	g[x].push_back(y);
	}
	ll ans=0;
    for(ll i=1;i<=n;++i){
        memset(sum,0,sizeof(sum));
        for(ll j=0;j<g[i].size();++j){
            ll u=g[i][j];
            for(ll k=0;k<g[u].size();++k){
                if(g[u][k]!=i) sum[g[u][k]]++;
            }
        }
        for(ll i=1;i<=n;++i) ans+=C(sum[i]);
    }
	cout<<ans<<"\n";
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_73654895/article/details/136782173