CodeChef TechFest 2013 Cool Numbers(搜索)

题目链接：http://www.codechef.com/INSCRP13/problems/INSCTS4

明天早上还有课，所以先刷两道吧。写完解题报告就睡觉！

这题是简单广度优先搜索，主要是如何建图，把数字在哪一行看作是边，数字和行看作是点，这么一分析立马就能AC了。

首先第一步是建图，我用了两个邻接表（一个也可以）个人认为两个结构清晰一些num对应数字在哪一行，row对应行上有哪些数字。

剩下的就是简单的对于每次查询从源点（查询数字）开始BFS，搜索步数即为答案。

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;

vector<int> num[2010], row[110];
int n, m, r, tn, vis[2010][110];

struct VEX{
    int num, step;
};

VEX v;

void init()
{
    for(int i=0; i<2010; i++) num[i].clear();
    for(int i=0; i<110; i++) row[i].clear();
}

int BFS(int x)
{
    queue<VEX> q;
    v.num = x;
    v.step = 0;
    q.push(v);
    while(!q.empty())
    {
        VEX tv = q.front();
        q.pop();
        if(tv.num==42) return tv.step;
        for(vector<int>::iterator ita = num[tv.num].begin(); ita!=num[tv.num].end(); ++ita){
            if(vis[tv.num][*ita]==0){
                vis[tv.num][*ita] = 1;
                for(vector<int>::iterator itb = row[*ita].begin(); itb!=row[*ita].end(); ++itb){
                    if(vis[*itb][*ita]==0){
                        vis[*itb][*ita] = 1;
                        VEX pv;pv.num = *itb;
                        pv.step = tv.step+1;
                        q.push(pv);
                    }
                }
            }
        }
    }
    return -1;
}

int main()
{
//    freopen("in.txt", "r", stdin);


    while(scanf("%d%d%d", &n, &m, &r)!=EOF)
    {
        init();
        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                scanf("%d", &tn);
                row[i].push_back(tn);
                num[tn].push_back(i);
            }
        }
        for(int i=1; i<=r; i++){
            memset(vis, 0, sizeof vis);
            scanf("%d", &tn);
            printf("%d", BFS(tn));
            if(i!=r)printf(" ");
        }
        printf("
");
    }
    return 0;
}