标签:pre seq 祖先 printf 最低 pos else int 公共
https://www.acwing.com/problem/content/description/1638/
思路:
LCA(爬山法)+重建一个树(知道其两个排序),由于时间n可能比较大,而且这题时间限制比较紧张,所以最后离散化,尽量查询不用哈希,因为哈希的常数太大了。
#include <iostream>
#include <cstring>
#include <unordered_map>
#include <algorithm>
using namespace std;
const int N = 10010;
int m, n;
int in[N], pre[N], seq[N];
unordered_map<int, int> pos;
int p[N], depth[N];
int build(int il, int ir, int pl, int pr, int d)
{
int root = pre[pl];
int k = root;
depth[root] = d;
if (il < k) p[build(il, k - 1, pl + 1, pl + 1 + (k - 1 - il), d + 1)] = root;
if (k < ir) p[build(k + 1, ir, pl + 1 + (k - 1 - il) + 1, pr, d + 1)] = root;
return root;
}
int main()
{
cin >> m >> n;
for (int i = 0; i < n; i ++ )
{
cin >> pre[i];
seq[i] = pre[i];
}
sort(seq, seq + n);
for (int i = 0; i < n; i ++ )
{
pos[seq[i]] = i;
in[i] = i;
}
for (int i = 0; i < n; i ++ ) pre[i] = pos[pre[i]];
build(0, n - 1, 0, n - 1, 0);
while (m -- )
{
int a, b;
cin >> a >> b;
if (pos.count(a) && pos.count(b))
{
a = pos[a], b = pos[b];
int x = a, y = b;
while (a != b)
if (depth[a] < depth[b]) b = p[b];
else a = p[a];
if (a != x && a != y) printf("LCA of %d and %d is %d.\n", seq[x], seq[y], seq[a]);
else if (a == x) printf("%d is an ancestor of %d.\n", seq[x], seq[y]);
else printf("%d is an ancestor of %d.\n", seq[y], seq[x]);
}
else if (pos.count(a) == 0 && pos.count(b) == 0)
printf("ERROR: %d and %d are not found.\n", a, b);
else if (pos.count(a) == 0)
printf("ERROR: %d is not found.\n", a);
else
printf("ERROR: %d is not found.\n", b);
}
return 0;
}
标签:pre,seq,祖先,printf,最低,pos,else,int,公共 来源: https://www.cnblogs.com/xjtfate/p/16603445.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。