小心可能为环
话说PAT OJ好像有bug,因为我一开始提交的A错了案例7,改一下的B错案例8,再提交A竟然AC了。。但是再提交A仍然错了案例7.
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
struct Node
{
string adr;
int va;
string next;
Node(string a,int b,string c):adr(a),va(b),next(c) {}
Node() {}
bool operator < (const Node & b) const
{
return va<b.va;
}
};
map<string,Node>m;
vector<Node>nodes;
int main()
{
string s,a,c;
int b,n,k;
cin>>s>>n>>k;
bool rnd = 1; //是否为环
for(int i = 0; i<n; i++)
{
cin>>a>>b>>c;
if(c == "-1") rnd = 0;
m[a] = Node(a,b,c);
}
int i = 0;
while(s!="-1")
{
if(i == n) break;
nodes.push_back(m[s]);
s = m[s].next;
i++;
}
// for(int i = 0; i<nodes.size(); i++)
// {
// cout<<nodes[i].adr<<" "<<nodes[i].va<<" "<<nodes[i].next<<endl;
// }
// cout<<endl;
for(int i = 0; i<nodes.size(); i++)
{
if(i%k == k-1)
{
reverse(nodes.begin()+i-k+1,nodes.begin()+i+1);
}
}
for(int i = 0; i<nodes.size(); i++)
{
if(i%k == k-1)
{
string s = nodes[i-k+1].adr;
for(int j = i-k+1; j<i; j++)
{
nodes[j].next = nodes[j+1].adr;
}
if(i+1!=nodes.size()){
nodes[i].next = nodes[i+1].adr;
}
else
{
if(!rnd) nodes[i].next = "-1";
else nodes[i].next = s;
}
}
}
// cout<<endl;
for(int i = 0; i<nodes.size(); i++)
{
cout<<nodes[i].adr<<" "<<nodes[i].va<<" "<<nodes[i].next<<endl;
}
return 0;
}
/**
00100 6 0
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218
00100 6 2
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218
00100 2 2
00100 1 12309
12309 2 00100
**/