数据结构oj_10

移除序列指定位置元素（链表）

问题描述 :

使用带头结点的单链表编程：

一群学生排成一行，输入一个位置，将该位置的学生移除。

第一个学生的位置为1，第n个学生的位置为n。

输入说明 :

第一行输入学生信息：

第一个整数n(0<=n<=100)，表示共有n个学生，其后有n个整数，表示n个学生的学号

第二行及以后各行，每行输入一个整数，表示要移除的学生位置。

输出说明 :

每次移除一个学生后，在一行里输出完整的学号序列，以head开头，以tail结尾，学号之间以“–>”分隔（参见范例）。

如果需要移除学生的位置不合法，则输出“invalid”（不包括双引号）

#include <iostream>
using namespace std;
struct student {
	int num;
	struct student* next;
};
student* creat(student *head, int n) {
//	cout << "创建链表" << endl;
	student *p, *q;
	p = head;
	q = head;
	for (int j = 1; j <= n; j++) {
		q = new student;
		p->next = q;
		p = q;
		cin >> p->num;
	}
	q->next = NULL;
	return head;
}
student* input(student* head, int n, int m) {
	head->num = 1;//num置1表状态正常，置0后返回invalid
	student *p, *q;
	student* mid = new student;
	mid->num = n;
	p = head;
	q = p;
	if (m == 1) {
		//段首插入mid
		if (head->next) {
			mid->next = head->next;
			head->next = mid;
		}
		else {
			//仅有头节点
			head->next = mid;
			mid->next = NULL;
		}
		return head;
	}
	else {
		for (int i = 1; i < m; i++) {
			//插入段向后移动并判断是否为段尾
			q = p;
			if (p->next == NULL) {
				head->num = 0;
				//仅有头节点且不在首位插入num置0
				return head;
			}
			p = p->next;
			if (p->next == NULL) {
				if (i < m - 1) {
					//在段尾且还需移动num置0
					head->num = 0;
					return head;
				}
				else {
					p->next = mid;
					mid->next = NULL;
					return head;
				}
			}
		}
		//中间插入mid
		p = p->next;
		q = q->next;
		mid->next = q->next;
		q->next = mid;
		return head;
	}
}

student* del(student* head, int n) {
//	cout << "删除节点" << endl;
	head->num = 1;
	student *p, *q;
	p = head;
	q = head;
	for (int i = 0; i < n; i++) {
		q = p;
		p = p->next;
		if (q->next == NULL) {
			head->num = 0;
			return head;
		}
		if (p->next == NULL) {
			if (i < n - 1) {
				head->num = 0;
				return head;
			}
			else {
				q->next = NULL;
				delete p;
				return head;
			}
		}
	}
	q->next = p->next;
	delete p;
	return head;
}


//head中值作为链表状态表示，实际使用中不对head做实际操作与显示
int main() {
	student* head = new student;
	int m,n, i;
	cin >> i;
	if (i == 0) {
		head->next = NULL;
	}
	else {
		head = creat(head, i);
	}
	while (cin >> m ) {
		if (m <= 0) {
			cout << "invalid" << endl;
		}
		else {
			head = del(head, m);
//			head = input(head, n, m);
			if (head->num != 0) {//判断num读数
				cout << "head-->";
				student *p = head;
				while (p->next != NULL) {
					p = p->next;
					cout << p->num << "-->";
				} 
				cout << "tail" << endl;
			}
			else {
				cout << "invalid" << endl;
			}
		}
	}
	return 0;
}

因为今天已经到了发布文章限制（10篇），剩下的明天再发吧