这个题确实没想到用bfs进行分层处理,后来看到了大佬的题解之后才想到了这一点
bfs dfs早就学了,可是还是不大会应用到这上面
可以分为三层,起始点,中间点,尾点,需要的数据是中间点到尾点的访问次数,这样就能得出形成的三角形了
因为是单向所以也不必考虑重复的问题
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn = 30005;
struct node{
int to,pre;
}edge[maxn];
int cnt,id[maxn];
void init()
{
cnt = 0;
memset(id,-1,sizeof(id));
}
void add(int s,int t)
{
edge[cnt] = (node){t,id[s]};
id[s] = cnt++;
}
int n,ans;
void bfs(int x)
{
int vis[3005];
memset(vis,0,sizeof(vis));
queue<int>q;
for(int i = id[x];~i;i = edge[i].pre)
q.push(edge[i].to);//传入第一层的点所指向的第二层点
while(q.size())
{
int t = q.front();
q.pop();
for(int i = id[t];~i;i = edge[i].pre)//查看第二层的点所指向的第三层的点
{
if(edge[i].to != x) vis[edge[i].to]++;//记录第三层点的被指向次数
}
}
for(int i = 1;i <= n;i++)//对于第三层的点被指向两次肯定形成1个,指向三次形成3个小小大依次类推即可
if(vis[i] >= 2)ans += vis[i] * (vis[i] - 1) / 2;
}
int main()
{
int m;
scanf("%d%d",&n,&m);
ans = 0;
init();
while(m--)
{
int s,t;
scanf("%d %d",&s,&t);//添加单向边
add(s,t);
}
for(int i=1;i<=n;i++) bfs(i);
printf("%d\n",ans);
return 0;
}