lightoj 1111 - Best Picnic Ever（dfs or bfs）

 題目連結 http://www.lightoj.com/volume_showproblem.php?problem=1111

題意：給你一個有向圖再給你幾個人的位置，問所有人可以在哪些點相聚。

簡單的搜尋題，可以用bfs也可以用dfs，要注意的是存邊的時候最好用vector，因為邊比較少。

用struct會逾時。下面附上dfs和bfs代碼。

#include <iostream>
#include <cstring>
#include <queue>
#include <vector>
#include <cstdio>
using namespace std;
int a[1100] , map[1100] , vis[1100];
vector<int> s[1100];
void bfs(int st) {
    queue<int>q;
    q.push(st);
    vis[st] = 1;
    map[st]++;
    while(!q.empty()) {
        int gg = q.front();
        int len = s[gg].size();
        for(int i = 0 ; i < len ; i++) {
            if(vis[s[gg][i]] != 1) {
                map[s[gg][i]]++;
                vis[s[gg][i]] = 1;
                q.push(s[gg][i]);
            }
        }
        q.pop();
    }
}
int main()
{
    int t;
    scanf("%d" , &t);
    int ans = 0;
    while(t--) {
        ans++;
        int k , n , m;
        for(int i = 0 ; i <= 1100 ; i++) {
            s[i].clear();
        }
        memset(map , 0 , sizeof(map));
        scanf("%d%d%d" , &k , &n , &m);
        for(int i = 0 ; i < k ; i++)
            scanf("%d" , &a[i]);
        for(int i = 0 ; i < m ; i++) {
            int x , y;
            scanf("%d%d" , &x , &y);
            s[x].push_back(y);
        }
        for(int i = 0 ; i < k ; i++) {
            memset(vis , 0 , sizeof(vis));
            bfs(a[i]);
        }
        int cnt = 0;
        for(int i = 1 ; i <= n ; i++) {
            //cout << map[i] << endl;
            if(map[i] == k)
                cnt++;
        }
        printf("Case %d: %d\n" , ans , cnt);
    }
    return 0;
}
           

#include <iostream>
#include <cstring>
#include <queue>
#include <vector>
#include <cstdio>
using namespace std;
int a[1100] , map[1100] , vis[1100];
vector<int> s[1100];
void dfs(int st) {
    vis[st] = 1;
    int len = s[st].size();
    for(int i = 0 ; i < len ; i++) {
        if(vis[s[st][i]] == 0) {
            map[s[st][i]]++;
            dfs(s[st][i]);
        }
    }
    return;
}
int main()
{
    int t;
    scanf("%d" , &t);
    int ans = 0;
    while(t--) {
        ans++;
        int k , n , m;
        for(int i = 0 ; i <= 1100 ; i++) {
            s[i].clear();
        }
        memset(map , 0 , sizeof(map));
        scanf("%d%d%d" , &k , &n , &m);
        for(int i = 0 ; i < k ; i++)
            scanf("%d" , &a[i]);
        for(int i = 0 ; i < m ; i++) {
            int x , y;
            scanf("%d%d" , &x , &y);
            s[x].push_back(y);
        }
        for(int i = 0 ; i < k ; i++) {
            memset(vis , 0 , sizeof(vis));
            map[a[i]]++;
            dfs(a[i]);
        }
        int cnt = 0;
        for(int i = 1 ; i <= n ; i++) {
            //cout << map[i] << endl;
            if(map[i] == k)
                cnt++;
        }
        printf("Case %d: %d\n" , ans , cnt);
    }
    return 0;
}
           

轉載于:https://www.cnblogs.com/TnT2333333/p/6078000.html