Uva 12569 Planning mobile robot on Tree (EASY Version)

基本思路就是Bfs：

　　本题的一个关键就是如何判段状态重复。

　　1.如果将状态用一个int型数组表示，即假设为int state[17]，state[0]代表机器人的位置，从1到M从小到大表示障碍物的位置。那么如果直接用STL中的set是会超时的，但如果自己建立一个hash方法，像这样：　　　

int getKey(State& s) {
    long long v = 0;
    for(int i=0; i<=M; ++i ) {
        v = v * 10 + s[i];
    }
    return v % hashSize;
}

 是可以避免超时的。

　　2. 除了上面这种，还有一种方法是用二进制法，因为有15个位置，除了robot，每个位置的状态只有 "有"或"无"，可以分别用1或0表示，那么就是15位的二进制，如

　　　　00000 00000 00001 可以表示位置0存在一个障碍物， 转化为int型整数即为 1

　　　　11111 00000 11111 可以表示位置0～4， 10～14各存在一个障碍物，转化为int型整数即为31775

　　　　那么所有障碍物的状态可以在0～2^15-1的范围内表示出来，另外单独考虑robot的位置，可以用一个bool数组，如 bool vis[32767][15]表示

　　　   每次移动障碍物时，先确定移动的起始位置（二进制为1的位置），再确定目标位置（二进制为0且与robot不重合的位置）。

　　　　同样机器人移动也是类似的道理。

　　　　因为采用了位运算，而且判重方法比上一种相对来得简单，所以速度要比上一种方法快一点。

 

PS：做这题时，刚开始采用了STL的set方法，结果超时。之后采用了自建hash表的方法，过了，但时间处于5000ms这个级别。后来观察别人的代码，才发现别人采用了二进制的方法(点击查看)，自己才恍然大悟，原理还可以这么做！！！于是自己琢磨了一下，写了一下，并且AC，发现速度大增，马上处于1000ms的级别。

 

#include <bits/stdc++.h>

using namespace std;

const int MAXN = (1 << 15) + 50;
bool vis[MAXN][15+2];
struct Node{
    int robot, ob, step;
};
typedef pair<int, int> Move; // from first to second
Node q[MAXN*17]; // queue
Move move[MAXN*17];
int movePath[MAXN*17]; // the step before rear is movePath[rear]
vector <int> link[17]; // link table
int n, m, s, target;
int start;

void Read() {
    cin >> n >> m >> s >> target;
    start = 0;
    s --;
    target --;
    int tmp;
    for(int i=0; i<m; ++i) {
        cin >> tmp;
        start = start | (1 << (tmp - 1)); // add ob at position tmp-1
    }
    for(int i=0; i<n; ++ i) {
        link[i].clear();
    }
    int x, y;
    for(int i=0; i<n-1; ++i) {
        cin >> x >> y;
        link[x-1].push_back(y-1);
        link[y-1].push_back(x-1);
    }
}

// print path
void Print(int rear) {
    vector<Move> v;
    while(rear) {
        v.push_back(move[rear]);
        rear = movePath[rear];
    }
    vector<Move>::iterator it;
    for(it = v.end()-1; it>=v.begin(); it--) {
        // +1 !!!
        cout << it->first + 1 << " " << it->second + 1 << endl;
    }
}

void Bfs() {
    int front = 0, rear = 1;
    q[front].ob = start;
    q[front].robot = s;
    q[front].step = 0;
    memset(vis, false, sizeof(vis));
    vis[q[front].ob][q[front].robot] = true;
    while(front < rear) {
        Node& t = q[front];
        if(t.robot == target) {
            cout << t.step << endl;
            Print(front);
            return ;
        }
        for(int i=0; i<n; ++i) { // move i
            if((t.ob & (1 << i)) || i == t.robot) { 
            // there is an ob or a robot at the position
                for(size_t j=0; j<link[i].size(); ++ j) {
                    int nextNode = link[i][j];
                    if(nextNode == t.robot || ((1<<nextNode) & t.ob)!=0 ) {
                        // there exists an ob or a robot at the next position
                        continue;
                    }
                    Node next = t;
                    if(i == t.robot) {
                        next.robot = nextNode;
                    } else {
                        // move from i to nextNode
                        next.ob = next.ob - (1 << i);
                        next.ob = next.ob + (1 << nextNode);
                    }
                    next.step = t.step + 1;
                    if(!vis[next.ob][next.robot]) {
                        vis[next.ob][next.robot] = true;
                        move[rear].first = i; // from
                        move[rear].second = nextNode; // next
                        movePath[rear] = front; // the former step
                        q[rear ++] = next; // push into the queue
                    }
                }
            }
        }
        ++ front;
    }
    cout << -1 << endl;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T, Case = 0;
    cin >> T;
    while(T --) {
        Read();
        cout << "Case " << (++Case) << ": ";
        Bfs();
    }
    return 0;
}