826. 单链表

Posted 2021-05-20 幽殇默

https://www.acwing.com/problem/content/description/828/
下标从0开始

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+10;
int head,e[N],ne[N],idx;
int n;
void init()
{
	head=-1;
	idx=0;
}
void to_add(int k,int x)
{
	e[idx]=x,ne[idx]=ne[k],ne[k]=idx++;
}
void add(int x)
{
	e[idx]=x,ne[idx]=head,head=idx++;
}
void remove(int k) 
{
	ne[k]=ne[ne[k]];
} 
int main(void)
{
	cin>>n;
	init();
	while(n--)
	{
		string op; cin>>op;	
		if(op=="H") 
		{
			int x; cin>>x;
			add(x);
		}
		if(op=="D")
		{
			int k; cin>>k;
			if(k) remove(k-1);
			else head=ne[head];
		}
		if(op=="I")
		{
			int k,x; cin>>k>>x;
			to_add(k-1,x);
		}
	}
	for(int i=head;i!=-1;i=ne[i])
	{
		cout<<e[i]<<" ";
	}
	return 0;
}

下标从1开始

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+10;
int head,e[N],ne[N],idx;
int n;
void init()
{
	head=-1;
	idx=1;
}
void to_add(int k,int x)
{
	e[idx]=x,ne[idx]=ne[k],ne[k]=idx++;
}
void add(int x)
{
	e[idx]=x,ne[idx]=head,head=idx++;
}
void remove(int k) 
{
	ne[k]=ne[ne[k]];
} 
int main(void)
{
	cin>>n;
	init();
	while(n--)
	{
		string op; cin>>op;	
		if(op=="H") 
		{
			int x; cin>>x;
			add(x);
		}
		if(op=="D")
		{
			int k; cin>>k;
			if(k) remove(k);
			else head=ne[head];
		}
		if(op=="I")
		{
			int k,x; cin>>k>>x;
			to_add(k,x);
		}
	}
	for(int i=head;i!=-1;i=ne[i])
	{
		cout<<e[i]<<" ";
	}
	return 0;
}