我的Data Structures and Algorithms代码仓:https://github.com/617076674/Data-Structures-and-Algorithms
原题链接:https://pintia.cn/problem-sets/16/problems/680
题目描述:
知识点:哈希表、拓扑排序
思路:拓扑排序
解决哈希冲突的线性探测法:
当得到key的hash值H(key),但是表中下标为H(key)的位置已经被某个其他元素使用了,那么就检查下一个位置H(key) + 1是否被占,如果没有,就使用这个位置;否则就继续检查下一个位置(也就是将hash值不断加1)。如果检查过程中超过了表长,那么就回到表的首位继续循环,直到找到一个可以使用的位置,或者是发现表中所有位置都已被使用。显然,这个做法容易导致扎堆,即表中连续若干个位置都被使用,这在一定程序上会降低效率。
计算key = nums[i] % N。
如果说key == i,即i节点就在它原本应该插入的位置。
如果nums[i] < 0,这种情况我们不考虑。
剩下一种情况,key != i,那么我们从key位置开始寻找,逐次递增1,寻找N次,一旦找到了nums[key] == nums[i],就说明我们找到了nums[i]所在的位置,跳出循环。否则,在(key + j) % N节点的后继节点中添加点i,即节点i的添加一定在这些节点之后。
再计算各个节点的入度,拓扑排序输出即可。由于本题要求优先输出较小的数,因此我们需要用一个优先队列,并且重载int型的比较函数cmp,令其按照nums[i]的大小来比较。
时间复杂度和空间复杂度均是O(N)
C++代码:
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
int N;
int nums[1000], inDegree[1000];
vector<int> graph[1000];
vector<int> result;
struct cmp{
bool operator() (int a, int b){ //重载int型的比较方法
return nums[a] > nums[b];
}
};
int main(){
fill(inDegree, inDegree + N, 0); //所有节点的入度初始化为0
scanf("%d", &N);
for(int i = 0; i < N; i++){
scanf("%d", &nums[i]);
if(nums[i] < 0){
inDegree[i] = -1; //不存在的点标记其入度为-1
}
}
for(int i = 0; i < N; i++){
int key = nums[i] % N; //key位置是nums[i]应该原本应该插入的位置
if(key == i || nums[i] < 0){
continue;
}
for(int j = 0; j < N; j++){ //从key位置递增,依次寻找节点i的所有前驱节点
if(nums[(key + j) % N] == nums[i]){
break;
}
graph[(key + j) % N].push_back(i);
}
}
for(int i = 0; i < N; i++){
for(int j = 0; j < graph[i].size(); j++){ //对每个点计算其入度值
inDegree[graph[i][j]]++;
}
}
priority_queue<int, vector<int>, cmp> q; //优先队列,采用自定义cmp函数
for(int i = 0; i < N; i++){ //将入度为0的点入队
if(inDegree[i] == 0){
q.push(i);
}
}
while(!q.empty()){
int u = q.top();
result.push_back(nums[u]);
q.pop();
for(int i = 0; i < graph[u].size(); i++){
int v = graph[u][i];
inDegree[v]--;
if(inDegree[v] == 0){
q.push(v);
}
}
}
for(int i = 0; i < result.size(); i++){
printf("%d", result[i]);
if(i != result.size() - 1){
printf(" ");
}else{
printf("\n");
}
}
return 0;
}
C++解题报告: