PAT_甲级_2020年冬季考试 7-3 File Path

乔梓鑫

7-3 File Path (25 分)

image.png
The figure shows the tree view of directories in Windows File Explorer. When a file is selected, there is a file path shown in the above navigation bar. Now given a tree view of directories, your job is to print the file path for any selected file.

Input Specification:

Each input file contains one test case. For each case, the first line gives a positive integer N (≤10^3), which is the total number of directories and files. Then N lines follow, each gives the unique 4-digit ID of a file or a directory, starting from the unique root ID 0000. The format is that the files of depth d will have their IDs indented by d spaces. It is guaranteed that there is no conflict in this tree structure.

Then a positive integer K (≤100) is given, followed by K queries of IDs.

Output Specification:

For each queried ID, print in a line the corresponding path from the root to the file in the format: 0000->ID1->ID2->...->ID. If the ID is not in the tree, print Error: ID is not found. instead.

Sample Input:

14
0000
 1234
  2234
   3234
    4234
    4235
    2333
   5234
   6234
    7234
     9999
  0001
   8234
 0002
4 9999 8234 0002 6666

Sample Output:

0000->1234->2234->6234->7234->9999
0000->1234->0001->8234
0000->0002
Error: 6666 is not found.

题目限制

image.png

题目大意

给定N个数字,每一个数字代表了文件或者文件夹,并且采用缩进的方式表示每一个文件或者文件夹之间的相对深度,缩进d格代表在d层。然后给定K个查询,要求输出从根节点到查询节点的路径,如果查询节点不存在输出Error: ID is not found.

算法思路

此题可以分为两步来进行,第一步建树,第二步深度优先搜索。难点就是建树,由于我们需要输出从根节点到查询节点的路径,并且字符串的长度就代表了节点的深度,那么我们直接使用一个二维数组,第一维代表了层数,第二维代表了在当前层数的所有节点,同时使用pre数组记录每一个节点的前驱节点。

  • 建树过程:我们使用字符串的长度size来代表节点的层数,那么每一次添加一个节点d,就记录当前节点的前驱节点,如果是第一个节点说明是根节点root,前驱为-1,否则前驱就是上一层中的最后一个节点 level [size - 1] [level [size - 1].size() - 1];

建树完毕后,对于每一个合法查询直接调用DFS进行深度搜索,从指定节点向前搜索,来到root后开始输出即可。

提交结果

image.png

AC代码

#include<cstdio>
#include<iostream>
#include<string>
#include<unordered_map>
#include<algorithm>
#include<vector>

using namespace std;

int pre[10004];// 记录每一个节点的前驱节点
unordered_map<int, bool> isExist;// 标记节点是否存在
vector<int> level[1005];// 每一层次的结点,按照顺序
int root;// 根节点

void DFS(int end) {
    if (root == end) {
        printf("%04d", end);
        return;
    }
    DFS(pre[end]);
    printf("->%04d", end);
}

int main() {
    int N;
    scanf("%d", &N);
    string current;
    // 吸收回车
    getchar();
    for (int i = 0; i < N; ++i) {
        getline(cin, current);
        int d = stoi(current);
        // 标记当前节点合法
        isExist[d] = true;
        // 使用字符串的长度代表层数,越小的层数越低
        int size = current.size();
        // 为当前层节点添加d
        level[size].push_back(d);
        // 记录每一个节点的前缀和根节点root
        if (i == 0) {
            pre[d] = -1;
            root = d;
        } else {
            // 每一个节点的前缀就是上一层节点的最后一个节点
            pre[d] = level[size - 1][level[size - 1].size() - 1];
        }
    }
    int K;
    scanf("%d", &K);
    for (int i = 0; i < K; ++i) {
        int a;
        scanf("%d", &a);
        if (!isExist[a]) {
            printf("Error: %04d is not found.\n", a);
        } else {
            DFS(a);
            printf("\n");
        }
    }
    return 0;
}
阅读 609

主要分享个人学习经验和心得

566 声望
15 粉丝
0 条评论

主要分享个人学习经验和心得

566 声望
15 粉丝
文章目录
宣传栏