老李在多年前承包了一个养猪场,并引入了若干只种猪,经过这些年的经营,现在养猪场有 N 只猪,编号从 0 到 N - 1(每只猪无论生死都有唯一的编号);
老李在每只猪生产的时候记下了生产的母猪和出生的小猪,格式:x y1 y2 y3 ...(注:x 为猪妈妈,y1,y2,y3 ... 为新生的猪仔,以上编码均在 0,...,N - 1内,每只猪可以多次生产,每个猪崽只有一个猪妈妈);
为了防疫需要,要检查任意两只猪是否有亲戚关系(两只猪具有相同的祖先),并计算关系亲疏情况(关系距离,相同编号距离为 0)
输入:第一行输入总数 N
第二行表示后续生产记录行数 M
后续 M 行输入生产记录,以空格分隔
最后一行输入m1,m2;表示待检查的m1和m2编号
输出
一个整数,表示m1,和m2之间的关系距离,无亲戚关系输出 -1
先根据输入 使用 unordered_map 记录每只小猪的猪妈妈是谁,并且找的树的根,将其妈妈编码设定为 -1,作为循环终止条件。
从 m1 开始遍历 m1 的所有祖先,若有 m2,输出当前距离,直接 return 掉;若不是 m2,则使用 unordered_map 把当前祖先到 m1 的距离记下来。
若没结束,再从 m2 开始遍历 m2 的所有祖先,若出现 m1 的祖先,直接将两个距离相加输出,结束。
如果此时还没结束,说明 m1 和 m2 没有公共祖先,则输出 -1,结束;
- #include
- #include
- using namespace std;
-
- int main() {
-
- int N, M;
- cin >> N >> M;
- unordered_map<int, int> pigs_mum;
- while (M--) {
- int pig_mum, pig_son;
- cin >> pig_mum;
- while (cin >> pig_son) {
- pigs_mum[pig_son] = pig_mum;
- if (cin.get() == '\n') break;
- }
- }
- for (int i = 0; i < N; ++i) {
- if (pigs_mum.find(i) == pigs_mum.end()) {
- pigs_mum[i] = -1;
- }
- }
-
- int m1, m2;
- cin >> m1 >> m2;
- unordered_map<int, int> distance;
- int tmp = m1, d = 0;
- while (tmp != -1) {
- if (tmp == m2) {
- cout << d << endl;
- return 0;
- }
- else {
- distance[tmp] = d;
- d++;
- tmp = pigs_mum[tmp];
- }
- }
- tmp = m2; d = 0;
- while (tmp != -1) {
- if (distance.find(tmp) != distance.end()) {
- cout << (d + distance[tmp]) << endl;
- return 0;
- }
- else {
- d++;
- tmp = pigs_mum[tmp];
- }
- }
- cout << -1 << endl;
-
- return 0;
- }
在一个 M * N 的街区种,有一个士兵 S 和一个敌人 E,表示 X 为无法通过的街区,标识 B 为可以通过的街区;士兵在一个单位时间内可以从一个街区移动到相邻的街区(士兵每次只能水平或者垂直方向移动一个街区);士兵每次改变方向时,需要额外花费一个单位的时间(士兵第一次移动一个街区时,不用考虑其初始方向,即只需要一个单位时间即可到达相邻街区)。计算士兵 S 最少 需要多少时间才能到达 E 所在的街区。
输入:第一行两个数组,表示街区大小,M行,N列;
接下来M行,每行N个字母,字母 S 表示士兵所在街区,字母 E 表示敌人所在街区,字母 X 表示障碍,字母 B 表示可以经过的街区。(只有一个 S,一个 E);
输出:最少需要的时间,当士兵 S 永远无法到达敌人 E 所在的街区时,输出 -1;
经典 dfs 题,笔试的时候脑袋抽筋了,搜到终点的时候,忘记 return 了,哎,难受!
- #include
- #include
- using namespace std;
-
- static int res = INT32_MAX;
- void dfs(vector
char >>& map, int i, int j, int step, int dir) { - if (i < 0 || i >= map.size() || j < 0 || j >= map[0].size()) return;
- if (map[i][j] == 'X' || map[i][j] == 'S') return;
- if (map[i][j] == 'E') {
- res = min(res, step);
- return;
- }
- map[i][j] = 'X';
- int step_plus[4];
- step_plus[0] = (dir == 0 ? 1 : 2);
- step_plus[1] = (dir == 1 ? 1 : 2);
- step_plus[2] = (dir == 2 ? 1 : 2);
- step_plus[3] = (dir == 3 ? 1 : 2);
- dfs(map, i - 1, j, step + step_plus[0], 0);
- dfs(map, i, j + 1, step + step_plus[1], 1);
- dfs(map, i + 1, j, step + step_plus[2], 2);
- dfs(map, i, j - 1, step + step_plus[3], 3);
- map[i][j] = 'B';
- }
-
- int main() {
-
- int m, n;
- cin >> m >> n;
- int start_x, start_y;
- vector
char>> map(m, vector<char>(n)); - for (int i = 0; i < m; ++i) {
- string s;
- cin >> s;
- for (int j = 0; j < n; ++j) {
- map[i][j] = s[j];
- if (s[j] == 'S') {
- start_x = i;
- start_y = j;
- }
- }
- }
-
- dfs(map, start_x - 1, start_y, 1, 0);
- dfs(map, start_x, start_y + 1, 1, 1);
- dfs(map, start_x + 1, start_y, 1, 2);
- dfs(map, start_x, start_y - 1, 1, 3);
-
- if (res == INT32_MAX) cout << -1 << endl;
- else cout << res << endl;
-
- return 0;
- }
某公司停车场安装了一套智慧停车系统,在车辆进入停车场时,系统会自动计算一条最短路径,并为这一条路径在地面点亮导路灯,如果没有可用停车位,则在入口亮起红灯。
为了简化处理,我们使用起始的车道坐标表示停车场入口,用一个二维数组来记录当前的停车情况,如:
- 8 15 8 1
- 1 2 2 1 2 2 1 2 2 1 2 2 1 2 1
- 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
- 1 2 1 2 1 0 1 3 2 2 2 1 2 2 2
- 1 2 1 2 2 0 1 3 1 2 1 2 1 2 1
- 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
- 2 1 2 2 2 0 2 2 1 1 0 1 2 1 2
- 2 2 2 2 2 0 2 2 2 2 0 2 2 2 2
- 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
请你根据停车场情况,点亮一条通往可用车位的最短路径。
路径规则:
输入:
M N i j
A11 ... A1j ... A1N
...
Ai1 ... Aij ... Ain
...
Am1 ... Amj ... Amn
输出:
输出导路灯的路径坐标
如果没有可用的停车位,则输出:-1 -1
如果存在可用的停车位,则输出最短路径(依次输出的每两个数代表一个点)
这就是一个 bfs 搜寻路径的题目,只是需要考虑的点较多;起点是给定的,终点未知,要在搜寻的过程中判断;
题目需要输出路径,所有需要维护一个二维 path 数组来记录路径,记录方式为将 path[i][j] 的值赋为 1, 2, 3, 4,分别表示当前点是从哪个方向搜过来的;
另外需要考虑路径距离相同的情况,题目给定有限考虑行标较小,或行和较小;
所以在 bfs 搜索每个节点的四个方向时,优先 上,在左,然后下,最后右,按照这个顺序去入队列,应该是可以保证最优的(笔试时细节没做好,没有运行出来,所以不确定)
仅代表个人想法,不确定是不是能 accept,题目所给测试案例都能通过;
- #include
- #include
- #include
- using namespace std;
-
- static int dir_x[] = {-1, 0, 1, 0};
- static int dir_y[] = {0, -1, 0, 1};
-
- int main() {
-
- int m, n, begin_x, begin_y;
- cin >> m >> n >> begin_x >> begin_y;
- vector
int>> parking(m, vector<int>(n)); - for (int i = 0; i < m; ++i) {
- for (int j = 0; j < n; ++j) {
- cin >> parking[i][j];
- }
- }
-
- begin_x--;
- begin_y--;
-
- vector
int>> path(m, vector<int>(n, 0)); - queue
int>> que; - que.push({begin_x, begin_y});
- parking[begin_x][begin_y] = 3;
- vector<int> end = {-1, -1};
- while (!que.empty()) {
- bool flag = false;
- int size = que.size();
- for (int i = 0; i < size; ++i) {
- vector<int> loc = que.front();
- que.pop();
- for (int i = 0; i < 4; ++i) {
- int next_x = loc[0] + dir_x[i], next_y = loc[1] + dir_y[i];
- if (next_x < 0 || next_x >= m || next_y < 0 || next_y >= n) continue;
- if ((i == 0 || i == 2) && parking[next_x][next_y] == 1) {
- end = {next_x, next_y};
- path[next_x][next_y] = i + 1;
- flag = true;
- break;
- }
- else if (parking[next_x][next_y] == 0) {
- que.push({next_x, next_y});
- parking[next_x][next_y] = 3;
- path[next_x][next_y] = i + 1;
- }
- }
- }
- if (flag) break;
- }
-
- if (end[0] == -1 && end[1] == -1) {
- cout << -1 << " " << -1 << endl;
- }
- else {
- vector<int> outPath;
- int x = end[0], y = end[1];
- while (x != begin_x || y != begin_y) {
- outPath.push_back(y);
- outPath.push_back(x);
- if (path[x][y] == 1) x--;
- if (path[x][y] == 2) y++;
- if (path[x][y] == 3) x++;
- if (path[x][y] == 4) y--;
- }
- outPath.push_back(begin_y);
- outPath.push_back(begin_x);
- for (int i = outPath.size() - 1; i > 0; --i) {
- cout << outPath[i] + 1 << " ";
- }
- cout << outPath[0] + 1 << endl;
- }
-
- return 0;
- }