数据结构oj_9

在序列指定位置插入元素（链表）

问题描述 :

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

一群学生排成一行，输入一个位置和一个学号，在该位置插入一个学生。

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

输入说明 :

第一行输入学生信息：

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

第二行及以后各行，每行输入两个整数，表示要插入的位置及学号。

输出说明 :

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

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

注：如果已有10个学生，插入在最前面，则插入位置为1，而要插入在最后面，则插入位置为11。

#include <iostream>
using namespace std;
struct student {
	int num;
	struct student* next;
};
student* creat(student *head, int n) {
	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;
	}
}

int main() {
	student* head = new student;
	int n, m, i;
	cin >> i;
	if (i == 0) {
		head->next = NULL;
	}
	else {
		head = creat(head, i);
	}
	while (cin >> m >> n) {
		if (m <= 0) {
			cout << "invalid" << endl;
		}
		else {
			head = input(head, n, m);
			if (head->num != 0) {//判断num读数
				cout << "head-->";
				student *p = head;
				do {
					p = p->next;
					cout << p->num << "-->";
				} while (p->next != NULL);
				cout << "tail" << endl;
			}
			else {
				cout << "invalid" << endl;
			}
		}
	}
	return 0;
}
			} while (p->next != NULL);
			cout << "tail" << endl;
		}
		else {
			cout << "invalid" << endl;
		}
	}
	return 0;
}