1075.链表元素分类(25)

PAT (Basic Level) Practise （中文）

1075.链表元素分类(25)

（这是我人生中第一篇CSDN博客，这个题17号考的，18号在寝室里请教了老师后AC的，考试的时候主要错误是STL erase的使用。还算新鲜，网上的分享应该还不多。）
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue
给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而[0, K]区间内的元素都排在大于K的元素前面。但每一类内部元素的顺序是不能改变的。例如：给定链表为 18→7→-4→0→5→-6→10→11→-2，K为10，则输出应该为 -4→-6→-2→7→0→5→10→18→11。

输入格式：

每个输入包含1个测试用例。每个测试用例第1行给出：第1个结点的地址；结点总个数，即正整数N (<= 105)；以及正整数K (<=1000)。结点的地址是5位非负整数，NULL地址用-1表示。

接下来有N行，每行格式为：

Address Data Next

其中Address是结点地址；Data是该结点保存的数据，为[-105, 105]区间内的整数；Next是下一结点的地址。题目保证给出的链表不为空。

输出格式：

对每个测试用例，按链表从头到尾的顺序输出重排后的结果链表，其上每个结点占一行，格式与输入相同。

输入样例：

00100 9 10
23333 10 27777
00000 0 99999
00100 18 12309
68237 -6 23333
33218 -4 00000
48652 -2 -1
99999 5 68237
27777 11 48652
12309 7 33218
输出样例：
33218 -4 68237
68237 -6 48652
48652 -2 12309
12309 7 00000
00000 0 99999
99999 5 23333
23333 10 00100
00100 18 27777
27777 11 -1

---

思路

不用结构体数组！提高编码效率！
利用一个vector V先实现静态链表。再利用一个vector ans存储要输出的结点。“使得所有负值元素都排在非负值元素的前面”，“[0, K]区间内的元素都排在大于K的元素前面”——整个程序被这3个条件分成了3个循环，每个循环内执行的语句都类似——遍历容器，将符合条件的结点压入ans。这样子是不会改变段内部的顺序的！得到的链表应该是【负数】+【0，k区间内的数】+【大于k的数】。维护好ans后，按照要求输出，注意ans[i].next=ans[i+1].address;，要更改结点的next信息！注意最后一个结点的next域是-1。

错误分析

考试的时候一直有stack的想法，觉得已经到ans的结点应该在V中删除，结果没有注意STL vector中的细节，出了错误，导致我第二个循环不能很好的执行if判断，在考场里百思不得其解，最后导致损失了很多分数，成绩不甚理想。
在一个用迭代器遍历容器的循环里，一定要注意删除erase方法的使用！！！所以在此我就不删除了，如果想要删除已经入ans的结点，应该使用如下代码：

	for (auto it = V.begin(); it<V.end();)
	{
		if (it->data >= 0 && it->data <= K)
		{
			ans.push_back(*it);
			it = V.erase(it);
		}
		else
		{
			it++;
		}
	}

以下为源代码

/*
  Name: 1075. 链表元素分类(25)
  Author: spencercjh
  Date: 2017年9月18日 08:26:56
  Description:PAT Basic Level C/C++ 9月17日考题
*/
#include<iostream>
#include<cmath>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<numeric>
using namespace::std;
struct node
{
  int address;
  int data;
  int next;
};
int main()
{
  int begin,N,K;
  node table[100000];
  scanf("%d %d %d",&begin,&N,&K);
  node* A=new node[N];
  vector<node> V;
  for(int i=0; i<N; i++)
  {
    scanf("%d %d %d",&A[i].address,&A[i].data,&A[i].next);
    table[A[i].address]=A[i];
  }
  for(int add=begin; add!=-1; add=table[add].next)
    V.push_back(table[add]);
  vector<node> ans;
  for(int i=0; i<V.size(); i++)
  {
    if(V[i].data<0)
    {
      ans.push_back(V[i]);
//      auto it=V.erase(V.begin()+i);	考试的时候执行了这2行语句，程序就不能有效执行了，去掉后就			     											
//      it=V.begin()+i;					AC，很遗憾……
    }
  }
  for(int i=0; i<V.size(); i++)
  {
    if(V[i].data>=0&&V[i].data<=K)
    {
      ans.push_back(V[i]);
//      auto it=V.erase(V.begin()+i);
//      it=V.begin()+i;
    }
  }
  for(int i=0; i<V.size(); i++)
  {
    if(V[i].data>K)
    {
      ans.push_back(V[i]);
//      auto it=V.erase(V.begin()+i);
//      it=V.begin()+i;
    }
  }
  for(int i=0; i<ans.size(); i++)
  {
    if(i!=ans.size()-1)
    {
      ans[i].next=ans[i+1].address;
//      cout<<ans[i].address<<" "<<ans[i].data<<" "<<ans[i].next<<endl;//  输出五位补0的数字
      printf("%05d %d %05d\n",ans[i].address,ans[i].data,ans[i].next);
    }
    else
//      cout<<ans[i].address<<" "<<ans[i].data<<" "<<-1<<endl;
      printf("%05d %d %d\n",ans[i].address,ans[i].data,-1);
  }
  return 0; 
}