题目描述
存在一个由 n
个不同元素组成的整数数组 nums
,但你已经记不清具体内容。好在你还记得 nums
中的每一对相邻元素。
给你一个二维整数数组 adjacentPairs ,大小为 n - 1 ,其中每个 adjacentPairs[i] = [ui, vi] 表示元素 ui 和 vi 在 nums 中相邻。
题目数据保证所有由元素 nums[i]
和 nums[i+1]
组成的相邻元素对都存在于 adjacentPairs
中,存在形式可能是 [nums[i], nums[i+1]]
,也可能是 [nums[i+1], nums[i]]
。这些相邻元素对可以 按任意顺序 出现。
返回 原始数组 nums
。如果存在多种解答,返回 其中任意一个 即可。
样例
输入:adjacentPairs = [[2,1],[3,4],[3,2]]
输出:[1,2,3,4]
解释:数组的所有相邻元素对都在 adjacentPairs 中。
特别要注意的是,adjacentPairs[i] 只表示两个元素相邻,并不保证其 左-右 顺序。
输入:adjacentPairs = [[4,-2],[1,4],[-3,1]]
输出:[-2,4,1,-3]
解释:数组中可能存在负数。
另一种解答是 [-3,1,4,-2] ,也会被视作正确答案。
输入:adjacentPairs = [[100000,-100000]]
输出:[100000,-100000]
提示:
nums.length == n
adjacentPairs.length == n - 1
adjacentPairs[i].length == 2
2 <= n <= 10^5
-10^5 <= nums[i], ui, vi <= 10^5
题目数据保证存在一些以 adjacentPairs
作为元素对的数组 nums
算法分析
图论
- 1、若数值
a
和数值b
相邻,则构造一条a
连接b
的边,枚举整个数组,进行建图 - 2、找到只出现一次的数值出发,枚举整个图,记录路径中的值即可
注意:
- 由于
2 <= n <= 10^5
,因此有10^5
个点,都是双向边,因此最多2 * 10^5
条边 - 由于
-10^5 <= nums[i], ui, vi <= 10^5
,数值有正有负,统一对每个数值添加B = 10^5
的偏移量全部变成正数,方便处理,因此数值范围是[0, 2 * 10^5]
,最后记录的时候再统一减去B = 10^5
即可
时间复杂度 $O(n)$
C++ 代码
const int N = 200010, B = 100000;
int h[N], e[N], ne[N], idx = 0;
int cnt[N];
class Solution {
public:
vector<int> path;
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void dfs(int u, int fa)
{
path.push_back(u - B);
for(int i = h[u]; i != -1;i = ne[i])
{
int j = e[i];
if(j == fa) continue;
dfs(j, u);
}
}
vector<int> restoreArray(vector<vector<int>>& ad) {
int n = ad.size();
memset(h, -1, sizeof h);
memset(cnt, 0, sizeof cnt);
idx = 0;
for(int i = 0;i < n;i ++)
{
int a = ad[i][0] + B, b = ad[i][1] + B;
add(a, b), add(b, a);
cnt[a] ++, cnt[b] ++;
}
int root = 0;
for(int i = 0;i < N;i ++)
if(cnt[i] == 1)
{
root = i;
break;
}
dfs(root, -1);
return path;
}
};
想到了是拓扑排序,可是卡在建图上了,想用邻接表,但是有负数,就卡死了,菜狗的悲哀啊!
大神!