1 题意
给定一个链表,其节点的地址为
5
5
5位十进制数,且键值不超过
1
0
4
10^{4}
104。
现要求对链表重组,键值绝对值第一次出现的节点重组为一个链表,其他的节点重组为另外一个链表。
比如有如下一个链表:
重组后得到如下两个链表:
链接:link。
2 思路
将地址作为数组下标,并保存每个节点下一跳地址和键值。
遍历整个链表,并将键值的绝对值记录在
v
i
s
[
]
vis[]
vis[]数组里面,将第一次出现的节点放在
r
e
s
1
[
]
res1[]
res1[]数组里,将其他节点放在
r
e
s
2
[
]
res2[]
res2[]数组里。
2.1 时间复杂度分析
每个节点只需要访问一次 ,所以时间复杂度为 O ( n ) \mathcal{O}(n) O(n)。
2.2 实现
#include<iostream>
#include<cstdio>
using namespace std;
const int N=1e5+10;
struct Node{
int nxt,val;
}node[N];
struct Res{
int adrs,val;
}res1[N],res2[N];
int vis[N],cnt1,cnt2;
void print(const Res *res,int cnt){
for(int i=1;i<=cnt;i++){
printf("%05d %d ",res[i].adrs,res[i].val);
if(i==cnt) printf("-1\n");
else printf("%05d\n",res[i+1].adrs);
}
}
int main(){
int root,n;scanf("%d %d",&root,&n);
for(int i=1;i<=n;i++){
int adrs,val,nxt;scanf("%d %d %d",&adrs,&val,&nxt);
node[adrs].val=val;
node[adrs].nxt=nxt;
}
while(root!=-1){
int val=node[root].val;
if(val<0) val=-val;
if(!vis[val]) res1[++cnt1].adrs=root,res1[cnt1].val=node[root].val,vis[val]=1;
else res2[++cnt2].adrs=root,res2[cnt2].val=node[root].val;
root=node[root].nxt;
}
print(res1,cnt1);print(res2,cnt2);
return 0;
}