codeforce 489d bfs分层处理

这个题确实没想到用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
",ans);
    return 0;
}