【ACM模板】链式前向星 存图 枚举每个结点连接的点 dfs bfs

//链式前向星 存图 枚举每个结点连接的点 dfs 
//四个结点的有向完全图（例子）
#pragma GCC optimize("O3")
#include<bits/stdc++.h>
using namespace std;
#define SET0(a) memset(a,0,sizeof(a))
#define FOR(i,a,b) for(int i=(a);i<=(b);i++)
#define DWN(i,a,b) for(int i=(a);i>=(b);i--)
typedef long long ll;
const int M=200005;
const int N=10005;
struct edge{
    int to,w,next;
}e[M];
int tot;
int head[N];
void add(int u,int v,int w){e[tot].to=v;e[tot].w=w;e[tot].next=head[u];head[u]=tot++;}
void dadd(int u,int v,int w){add(u,v,w);add(v,u,w);}
void enum_point(int u){
    for(int i=head[u];~i;i=e[i].next){
        cout<<"to the "<<e[i].to<<" and wealth is "<<e[i].w<<' ';
        cout<<endl;
    }
}

bool vis[N];
void dfs(int st){
    cout<<st<<' ';
    for(int i=head[st];~i;i=e[i].next){
        int pt=e[i].to;
        if(vis[pt]) continue;
        
        vis[pt]=true;
        dfs(pt);
    }
}
void bfs(int st){
    queue<int>q;
    q.push(st);
    vis[st]=true;
    while(!q.empty()){
        int hd=q.front();
        q.pop();
        cout<<hd<<' ';
        for(int i=head[hd];~i;i=e[i].next){
            int pt=e[i].to;
            if(vis[pt]) continue;

            vis[pt]=true;
            q.push(pt);
        }
    }
}
int main(){
    memset(head,-1,sizeof(head));
    dadd(1,2,1); dadd(1,3,2); dadd(1,4,3);
    dadd(2,3,4); dadd(2,4,5);
    dadd(3,4,6);

    for(int i=1;i<=4;i++){
        cout<<"node "<<i<<" message:"<<endl;
        enum_point(i);
        cout<<endl;
    }
    

    vis[1]=1;
    dfs(1);
    cout<<endl;
    memset(vis,false,sizeof(vis));
    bfs(1);
    return 0;
}