POJ 3256 Cow Picnic

水BFS

#include <iostream>
#include <cstdio>
#include <vector>
#include <cstring>
#include <queue>
using namespace std;
int cnt[1010],a[120];
bool ck[1010][120];
vector<int> g[1010];
queue<int> q;
int main()
{
    int n,m,k,i,j,ans;
    scanf("%d%d%d",&n,&m,&k);
    memset(cnt,0,sizeof(cnt));
    memset(ck,false,sizeof(ck));
    for (i=0; i<1010; i++)
        g[i].clear();
    while (!q.empty())
        q.pop();
    for (i=1; i<=n; i++)
    {
        scanf("%d",a+i);
        cnt[a[i]]++;
    }
    while (k--)
    {
        scanf("%d%d",&i,&j);
        g[i].push_back(j);
    }
    for (i=1; i<=n; i++)
    {
        q.push(a[i]);
        ck[a[i]][i]=true;
        while (!q.empty())
        {
            j=q.front();
            q.pop();
            for (k=0; k<g[j].size(); k++)
            {
                if (ck[g[j][k]][i] == false)
                {
                    ck[g[j][k]][i]=true;
                    cnt[g[j][k]]++;
                    q.push(g[j][k]);
                }
            }
        }
    }
    ans=0;
    for (i=1; i<=m; i++)
    {
        if (cnt[i] == n)
            ans++;
    }
    printf("%d\n",ans);
}