给定一个带整数键值的链表 L,你需要把其中绝对值重复的键值结点删掉。即对每个键值 K,只有第一个绝对值等于 K 的结点被保留。同时,所有被删除的结点须被保存在另一个链表上。例如给定 L 为 21→-15→-15→-7→15,你需要输出去重后的链表 21→-15→-7,还有被删除的链表 -15→15。
输入格式:
输入在第一行给出 L 的第一个结点的地址和一个正整数 N(≤105,为结点总数)。一个结点的地址是非负的 5 位整数,空地址 NULL 用 -1 来表示。
随后 N 行,每行按以下格式描述一个结点:
地址 键值 下一个结点
其中
地址
是该结点的地址,键值
是绝对值不超过104的整数,下一个结点
是下个结点的地址。输出格式:
首先输出去重后的链表,然后输出被删除的链表。每个结点占一行,按输入的格式输出。
输入样例:
00100 5 99999 -7 87654 23854 -15 00000 87654 15 -1 00000 -15 99999 00100 21 23854
输出样例:
00100 21 23854 23854 -15 99999 99999 -7 -1 00000 -15 87654 87654 15 -1
思路:
使用不去重的map记录字符串对应的下一个字符串及其键值
struct node{
string ne ; int w;
};
unordered_map<string,node>mp;
for(int i = 1 ; i <= T ; i ++){
string a , b;int c;
cin >> a >> c >> b;
mp[a] = {b,c};
}
从第一个往后找,如果键值有重复或者出现过就将后一个放入vector中,输出这次的结果继续进行dfs;
void dfs(string s){
if(s == "-1")return;
string t = mp[s].ne;
while(t != "-1" && (abs(mp[t].w) == abs(mp[s].w)||
vis[abs(mp[t].w)])){
v.push_back(t);
t = mp[t].ne;
}
vis[abs(mp[s].w)] = 1;
cout << s << " " << mp[s].w << " " << t << endl;
dfs(t);
}
总代码:
#include<bits/stdc++.h>
using namespace std;
struct node{
string ne ; int w;
};
unordered_map<string,node>mp;
unordered_map<int,int>vis;
vector<string>v;
void dfs(string s){
if(s == "-1")return;
string t = mp[s].ne;
while(t != "-1" && (abs(mp[t].w) == abs(mp[s].w)||
vis[abs(mp[t].w)])){
v.push_back(t);
t = mp[t].ne;
}
vis[abs(mp[s].w)] = 1;
cout << s << " " << mp[s].w << " " << t << endl;
dfs(t);
}
int main(){
ios::sync_with_stdio(false);
string s; int T;
cin >> s >> T;
for(int i = 1 ; i <= T ; i ++){
string a , b;int c;
cin >> a >> c >> b;
mp[a] = {b,c};
}
dfs(s);
for(int i = 0 ; i < v.size() ; i ++){
cout << v[i] << " " << mp[v[i]].w << " " ;
if(i != v.size() - 1)cout << v[i+1] << endl;
else cout << "-1" << endl;
}
}