Pat1025代码
题目描述:
给定一个常数K以及一个单链表L,请编写程序将L中每K个结点反转。例如:给定L为1→2→3→4→5→6,K为3,则输出应该为3→2→1→6→5→4;如果K为4,则输出应该为4→3→2→1→5→6,即最后不到K个元素不反转。
输入格式:
每个输入包含1个测试用例。每个测试用例第1行给出第1个结点的地址、结点总个数正整数N(<= 105)、以及正整数K(<=N),即要求反转的子链结点的个数。结点的地址是5位非负整数,NULL地址用-1表示。
接下来有N行,每行格式为:
Address Data Next
其中Address是结点地址,Data是该结点保存的整数数据,Next是下一结点的地址。
输出格式:
对每个测试用例,顺序输出反转后的链表,其上每个结点占一行,格式与输入相同。
输入样例:00100 6 4 00000 4 99999 00100 1 12309 68237 6 -1 33218 3 00000 99999 5 68237 12309 2 33218输出样例:
00000 4 33218 33218 3 12309 12309 2 00100 00100 1 99999 99999 5 68237 68237 6 -1
#include<cstdio>
#include<cstring>
#include<vector>
#define MAX 100001
using namespace std;
typedef struct Node
{
char addr[6];
int data;
char next[6];
}Node;
int visited[MAX];
int main(int argc,char *argv[])
{
int n,k;
int index,i,j;
char first[6];
vector<Node> v;
Node N[MAX];
scanf("%s %d %d",first,&n,&k);
for(i=0;i<n;i++)
scanf("%s %d %s",N[i].addr,&N[i].data,N[i].next);
for(i=0;i<n;i++)
if(strcmp(N[i].addr,first)==0)
{
index=i;
break;
}
for(i=0;i<n;i++)
{
v.push_back(N[index]);
visited[index]=1;
for(j=0;j<n;j++)
if(visited[j])
continue;
else
{
if(strcmp(N[index].next,N[j].addr)==0)
{
index=j;
break;
}
}
}
int loop=n/k;
for(i=1;i<=loop;i++)
{
j=i*k;
for(int m=1;m<=k;m++)
{
if(i<loop&&n%k==0&&m==k)
strcpy(v[j-1].next,v[(i+1)*k-1].addr);
else if(i==loop&&m==k)
{
if(n%k==0&&i==loop)
strcpy(v[j-1].next,"-1");
else if(n%k!=0)
strcpy(v[j-1].next,v[loop*k].addr);
}
else
strcpy(v[j-1].next,v[j-2].addr);
printf("%s %d %s\n",v[j-1].addr,v[j-1].data,v[j-1].next);
j--;
}
}
for(i=loop*k;i<n;i++)
printf("%s %d %s\n",v[i].addr,v[i].data,v[i].next);
return 0;
}
额,一个case超时,一个case错误,求更好的解题方法!!!
最近有人告诉我,最后一个case包含一些无效节点,最后改了改代码就AC了。。。
其实继续按照上面的那种方法,倒数第二个case超时,毕竟是O(n*n)的复杂度,最后想了一个用O(n)复杂度将链表连起来的方法。如果不将链表链接起来也是可以AC的,可以参照一楼的方法。(PS:Pat太坑。。。)
AC代码:
#include<cstdio>
#include<vector>
#define MAX 100005
using namespace std;
typedef struct Node
{
int addr;
int data;
int next;
}Node;
typedef struct Data
{
int data;
int next;
}Data;
int main(int argc,char *argv[])
{
int n,k,cur;
int index,i,j;
int first;
vector<Node> v(MAX);
Data D[MAX];
scanf("%d %d %d",&first,&n,&k);
for(i=0;i<n;i++)
{
int addr,data,next;
scanf("%d %d %d",&addr,&data,&next);
D[addr].data=data;
D[addr].next=next;
}
int size=0;
cur=first;
while(1)//将链表串起来
{
Node temp;
temp.addr=cur;
temp.data=D[cur].data;
temp.next=D[cur].next;
v[size]=temp;
size++;
if(D[cur].next==-1)
break;
cur=D[cur].next;
}
int loop=size/k;
for(i=1;i<=loop;i++)//反转
{
j=i*k-1;
for(int m=1;m<=k;m++)
{
if(i<loop&&m==k)//反转中间的K个元素,例如 1 2 3 4 5 6 7 ,k=3
v[j].next=v[(i+1)*k-1].addr;//一次循环后1指向6
else if(i==loop&&m==k)
{
if(size%k==0)//如果反转的是最后一个元素,next置-1
v[j].next=-1;
else//如果不能全部反转,next指向不用反转的地址
v[j].next=v[loop*k].addr;
}
else
v[j].next=v[j-1].addr;//反转,将next指向前面的地址
if(v[j].next==-1)
printf("%05d %d %d\n",v[j].addr,v[j].data,v[j].next);
else
printf("%05d %d %05d\n",v[j].addr,v[j].data,v[j].next);
j--;
}
}
for(i=loop*k;i<size;i++)
if(v[i].next==-1)
printf("%05d %d %d\n",v[i].addr,v[i].data,v[i].next);
else
printf("%05d %d %05d\n",v[i].addr,v[i].data,v[i].next);
return 0;
}