来自多个来源的BFS上的TLE

时间:2012-11-29 01:51:49

标签: c++ graph-theory breadth-first-search

我正在尝试解决这个问题:http://olimpiada-informatica.org/?cmd=downloadE&pbm=velo101&ext=pdf这是西班牙语,但我会尝试在此处翻译:

  

当你发现有人在着陆跑道上释放了一些迅猛龙时,你即将降落在地球上。
  着陆旅行具有矩形形状。我们用一个圆点(.)标记一个空白点,其中V为速龙,并且#点上有障碍物(你不能登陆它们)。
  速龙可以花一秒钟去另一个地方,但它们只能水平和垂直移动   系统会要求您标记X那些位于其上的点,您将最大限度地延长剩余使用寿命。

我已经做了一个算法,在这个算法中,我采用了速度计的每一个位置并在每个位置上制作了一个BFS,但是我得到了TLE,这是我的代码:

http://ideone.com/a6BVv3

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

int cost[501][501],xx,yy,n,m;
char mat[501][501];
bool visit[501][501],first = true;

int a[] = {-1,0,0,1}, b[] = {0,-1,1,0};

void check(int x,int y,int level) {
  cost[x][y] = level;
  for(int i = 0; i < 4; ++i) {
    xx = x + a[i];
    yy = y + b[i];
    if(0 <= xx and xx < n and 0 <= yy and yy < m and mat[xx][yy] == '.') {
      if(!visit[xx][yy] or level + 1 < cost[xx][yy]) {
        visit[xx][yy] = true;
        check(xx,yy,level + 1);
      }
    }
  }
}

int max() {
  int r = -1;
  for(int i = 0; i < n; ++i) for(int j = 0; j < m; ++j) if(mat[i][j] == '.') r = max(r,cost[i][j]);
  return r;
}

void show() {
  if(!first) puts("---");
  int r = max();
  for(int i = 0; i < n; ++i) {
    for(int j = 0; j < m; ++j) {
      if(cost[i][j] == r) printf("X");
      else printf("%c",mat[i][j]);
    }
    puts("");
  }
}

int main() {
  while(scanf("%d %d",&n,&m) == 2) {
    queue<pair<int,int> > cola;
    for(int i = 0; i < n; ++i) {
      scanf("\n");
      for(int j = 0; j < m; ++j) {
        scanf("%c",&mat[i][j]);
        if(mat[i][j] == 'V') cola.push(make_pair(i,j));
      }
    }
    memset(cost,-1,sizeof cost);
    memset(visit,0,sizeof visit);
    while(!cola.empty()) {
      pair<int,int> aux = cola.front();
      visit[aux.first][aux.second] = true;
      check(aux.first, aux.second,0);
      cola.pop();
    }
    show();
    first = false;
  }
  return 0;
}

有谁知道如何改进我的算法?

修改

好的,我能够解决问题,如果有人有兴趣,这里是代码:

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

int cost[501][501],n,m;
char mat[501][501];
bool visit[501][501],first = true;
queue<pair<int,int> > cola;

int a[] = {-1,0,0,1}, b[] = {0,-1,1,0};

int max() {
  int r = -1;
  for(int i = 0; i < n; ++i) for(int j = 0; j < m; ++j) if(mat[i][j] == '.') r = max(r,cost[i][j]);
  return r;
}

void show() {
  if(!first) puts("---");
  int r = max();
  for(int i = 0; i < n; ++i) {
    for(int j = 0; j < m; ++j) {
      if(cost[i][j] == r) printf("X");
      else printf("%c",mat[i][j]);
    }
    puts("");
  }
}

int main() {
  int cont = 0,x,y,xx,yy,level;
  while(scanf("%d %d",&n,&m) == 2) {
    for(int i = 0; i < n; ++i) {
      scanf("\n");
      for(int j = 0; j < m; ++j) {
        scanf("%c",&mat[i][j]);
        if(mat[i][j] == 'V') cola.push(make_pair(i,j));
      }
    }
    memset(cost,-1,sizeof cost);
    memset(visit,0,sizeof visit);
    while(!cola.empty()) {
      int s_cola = cola.size();
      for(int i = 0; i < s_cola; ++i) {
        x = cola.front().first, y = cola.front().second;
        cola.pop();
        level = cost[x][y];
        for(int i = 0; i < 4; ++i) {
          xx = x + a[i], yy = y + b[i];
          if(0 <= xx and xx < n and 0 <= yy and yy < m and mat[xx][yy] == '.') {
            if(!visit[xx][yy] or level + 1 < cost[xx][yy]) {
              visit[xx][yy] = true;
              cost[xx][yy] = level + 1;
              cola.push(make_pair(xx,yy));
            }
          }
        }
      }
    }
    show();
    first = false;
  }
  return 0;
}

1 个答案:

答案 0 :(得分:1)

您正在check()中对整个图形进行深度优先搜索。将它与main()中的循环集成,而不是尝试深度优先找到最短路径。