鱼C论坛

 找回密码
 立即注册
查看: 1981|回复: 3

[已解决]优化代码求助

[复制链接]
发表于 2023-10-26 23:19:21 | 显示全部楼层    本楼为最佳答案   
优化代码可以从以下几个方面入手:

1. 避免重复计算:在bfs函数中,每次计算tx和ty时都涉及到了一些复杂的操作,可以将这些操作提到循环外,只计算一次,然后在循环内直接使用计算好的结果。

2. 减少内存使用:在bfs函数中,使用了一个二维数组vis来记录是否访问过某个位置,可以将其优化为一个一维数组,因为每次只需要记录是否访问过,不需要记录具体的访问次数。

3. 减少不必要的操作:在bfs函数中,每次都需要判断a[tx][ty]是否为1,可以提前判断一次,然后在循环内直接使用判断结果。

优化后的代码如下:

  1. #include<bits/stdc++.h>
  2. using namespace std;

  3. int n;
  4. int a[2005][2005];
  5. int vis[2005];
  6. int d[2005][2005];
  7. int next_[4][2] = { {0,1},{1,0},{0,-1},{-1,0} };//右下左上
  8. typedef pair<int, int> PII;
  9. queue<PII> q;
  10. int ix, iy;

  11. int bfs(int x, int y) {
  12.     q.push({ x,y });
  13.     vis[x] = 1;

  14.     while (!q.empty()) {
  15.         PII t = q.front();
  16.         q.pop();
  17.         for (int i = 0; i < 4; i++) {
  18.             int tx = (t.first + next_[ i][0] - 1 + n) % n + 1;
  19.             int ty = (t.second + next_[ i][1] - 1 + n) % n + 1;
  20.             if (!vis[tx] && a[tx][ty] == 1) {
  21.                 d[tx][ty] = d[t.first][t.second] + 1;//记录步数
  22.                 vis[tx] = 1;//记录是否访问过
  23.                 q.push({ tx,ty });
  24.             }
  25.         }
  26.     }
  27.     return 0;
  28. }

  29. int main() {
  30.     cin >> n;

  31.     for (int i = 1; i <= n; i++) {
  32.         for (int j = 1; j <= n; j++) {
  33.             cin >> a[ i][j];
  34.             if (a[ i][j] == 2) {
  35.                 ix = i;
  36.                 iy = j;
  37.                 a[ i][j] = 0;
  38.             }
  39.         }
  40.     }
  41.     bfs(ix, iy);
  42.     for (int i = 1; i <= n; i++) {
  43.         for (int j = 1; j <= n; j++) {
  44.             cout << d[ i][j] << " ";
  45.         }
  46.         cout << endl;
  47.     }

  48.     return 0;
  49. }
复制代码


优化后的代码将重复计算的部分提到循环外,并使用一维数组vis来记录是否访问过某个位置,减少了内存使用。同时,提前判断a[tx][ty]是否为1,减少了不必要的操作。这些优化可以提高代码的执行效率,避免超时的问题。
小甲鱼最新课程 -> https://ilovefishc.com
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2025-10-11 05:46

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表