惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

IT之家
IT之家
The GitHub Blog
The GitHub Blog
F
Fortinet All Blogs
Last Week in AI
Last Week in AI
OSCHINA 社区最新新闻
OSCHINA 社区最新新闻
L
LangChain Blog
爱范儿
爱范儿
博客园_首页
Stack Overflow Blog
Stack Overflow Blog
MongoDB | Blog
MongoDB | Blog
博客园 - 三生石上(FineUI控件)
大猫的无限游戏
大猫的无限游戏
宝玉的分享
宝玉的分享
GbyAI
GbyAI
H
Help Net Security
A
About on SuperTechFans
Recent Announcements
Recent Announcements
Hugging Face - Blog
Hugging Face - Blog
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
雷峰网
雷峰网
D
Docker
博客园 - Franky
有赞技术团队
有赞技术团队
G
Google Developers Blog

OhYee 博客

小鹏辅助驾驶测评|OhYee 博客 小鹏非支持手机开启自动解锁|OhYee 博客 使用函数计算实现 301 重定向|OhYee 博客 针对 HTML 内容使用 Ant Design 图片弹框|OhYee 博客 博客进程泄露及僵尸进程解决|OhYee 博客 蓝易云服务器体验|OhYee 博客 SSH 调起本地 VSCode|OhYee 博客 【2022 秋招内推】阿里云后端研发工程师|OhYee 博客 使用函数计算获取 IP 地址信息|OhYee 博客 正确获取客户端 IP/HTTP Header 也可能重复|OhYee 博客 评测 Oculus Quest2 及 BigScreen|OhYee 博客 NextJS 热重载保留状态|OhYee 博客 如何优雅地贴 gist 代码|OhYee 博客 Linux 精细化文件权限|OhYee 博客 VSCode 容器开发环境|OhYee 博客 Clash 的不兼容更新排查|OhYee 博客 Zeek 导出 PCAP|OhYee 博客 记一次 ssh 配置问题|OhYee 博客 Git Commit 规范化工具|OhYee 博客 谈谈《星之卡比-探索发现》|OhYee 博客 VSCode 快捷键绑定 Shell 命令|OhYee 博客 ASN.1 语法及 X.509 证书格式解析解析|OhYee 博客 腾讯企业邮箱忽略 MX 记录发信|OhYee 博客 Chrome/Edge 标签组插件|OhYee 博客 【应届内推】阿里云后端研发工程师|OhYee 博客 损坏的 Typecho 备份处理为 JSON|OhYee 博客 VS Code VIM 插件高效使用|OhYee 博客 SSH 正反向代理|OhYee 博客 Let's Encrypt 根证书过期引发的问题|OhYee 博客 OpenWRT 忽略内核依赖|OhYee 博客
hihocoder 1490.Tree Restoration|OhYee 博客
2018-04-02 · via OhYee 博客

原题链接:http://hihocoder.com/problemset/problem/1490?sid=1299846
题目看上去很麻烦,先看样例
第一行:节点数N,层数M,叶子数K
第二行:M个数,表示每层的节点数A[I]
接下来M行,每行A[I]个数,表示该层节点编号B[I][j](从左到右顺序)
然后是一行K个数,表示叶子的编号
最后K×K的矩阵表示叶子间距离D[I][j]

根据两个节点间的距离,可以得知两个节点是否是同一个节点的子节点(距离为1)
由于给了节点顺序,因此我们只需要贪心地往左面的节点连接子节点即可

因此,问题转换成判断同一层相邻两个节点间的距离是否为2
如果为2,则接到同一个节点;否则,接到上一层的下一个非叶子节点上

至于距离,可以找到该节点下面的任意一个叶子节点,用这个叶子节点到目标的值减去层数差即可

#include <cstdio>
#include <cstring>
using namespace std;

#define Log(format, ...) //printf(format, ##__VA_ARGS__)

const int maxn = 105;

int nodeNum[maxn];
int nodeIdx[maxn][maxn];
int leaves[maxn];
bool isLeaf[maxn];
int dis[maxn][maxn];

int parents[maxn];
int child[maxn];
int delta[maxn];

int findAndUpdateLastNode(int &lastNode, int &lastNodeIdx, int deep,
                          int nowIdx) {
    // 查找并更新上一层的第一个有孩子的节点
    for (int k = lastNode+1; k < nodeNum[deep - 1]; ++k) {
        if (!isLeaf[nodeIdx[deep - 1][k]]) {
            lastNode = k;
            break;
        }
    }
    lastNodeIdx = nodeIdx[deep - 1][lastNode];

    child[lastNodeIdx] = child[nowIdx];
    delta[lastNodeIdx] = delta[nowIdx] + 1;

    return lastNode;
}

int main() {
    memset(parents, 0, sizeof(parents));
    memset(isLeaf, false, sizeof(isLeaf));
    memset(delta, 0, sizeof(delta));
    memset(parents, 0, sizeof(parents));
    memset(delta, 0, sizeof(delta));

    int N, M, K;
    scanf("%d%d%d", &N, &M, &K);
    for (int i = 0; i < M; ++i)
        scanf("%d", &nodeNum[i]);
    for (int i = 0; i < M; ++i)
        for (int j = 0; j < nodeNum[i]; ++j)
            scanf("%d", &nodeIdx[i][j]);
    for (int i = 0; i < K; ++i) {
        scanf("%d", &leaves[i]);
        isLeaf[leaves[i]] = true;
        child[leaves[i]] = i;
    }
    for (int i = 0; i < K; ++i)
        for (int j = 0; j < K; ++j)
            scanf("%d", &dis[i][j]);

    Log("input ok\n");

    for (int i = M - 1; i > 0; --i) { // 处理第i层
        Log("\tdepth:%d\n", i);
        int lastNode = -1;
        int lastNodeIdx = -1;

        for (int j = 0; j < nodeNum[i]; ++j) { // 处理当前层的第j个节点
            int nowIdx = nodeIdx[i][j];        // 当前节点的编号
            Log("\t\tnowIdx:%d\n", nowIdx);
            if (lastNode == -1) {
                // 处理第一个节点
                findAndUpdateLastNode(lastNode, lastNodeIdx, i, nowIdx);
                parents[nowIdx] = lastNodeIdx;
            } else {
                int preNodeIdx = nodeIdx[i][j - 1];
                Log("\t\t\tpreNodeIdx:%d\n", preNodeIdx);

                int distance = dis[child[nowIdx]][child[preNodeIdx]] -
                               delta[nowIdx] - delta[preNodeIdx];
                Log("\t\t\tdistance:%d dis:%d(%d %d) del:%d %d\n", distance,
                    dis[child[nowIdx]][child[preNodeIdx]], child[nowIdx],
                    child[preNodeIdx], delta[nowIdx], delta[preNodeIdx]);
                if (distance == 2) {
                    parents[nowIdx] = lastNodeIdx;
                } else {
                    findAndUpdateLastNode(lastNode, lastNodeIdx,
                                                     i, nowIdx);
                    parents[nowIdx] = lastNodeIdx;
                }
            }
            Log("\t\t\tlastNode:%d LastNodeIdx:%d\n", lastNode, lastNodeIdx);
        }
    }
    for (int i = 1; i <= N; ++i) {
        printf("%d ", parents[i]);
    }
    printf("\n");
    return 0;
}