mirror of
https://github.com/InternLM/InternBootcamp.git
synced 2026-04-23 16:55:02 +00:00
146 lines
4.6 KiB
Python
Executable file
146 lines
4.6 KiB
Python
Executable file
"""#
|
|
|
|
### 谜题描述
|
|
One day Bob got a letter in an envelope. Bob knows that when Berland's post officers send a letter directly from city «A» to city «B», they stamp it with «A B», or «B A». Unfortunately, often it is impossible to send a letter directly from the city of the sender to the city of the receiver, that's why the letter is sent via some intermediate cities. Post officers never send a letter in such a way that the route of this letter contains some city more than once. Bob is sure that the post officers stamp the letters accurately.
|
|
|
|
There are n stamps on the envelope of Bob's letter. He understands that the possible routes of this letter are only two. But the stamps are numerous, and Bob can't determine himself none of these routes. That's why he asks you to help him. Find one of the possible routes of the letter.
|
|
|
|
Input
|
|
|
|
The first line contains integer n (1 ≤ n ≤ 105) — amount of mail stamps on the envelope. Then there follow n lines with two integers each — description of the stamps. Each stamp is described with indexes of the cities between which a letter is sent. The indexes of cities are integers from 1 to 109. Indexes of all the cities are different. Every time the letter is sent from one city to another, exactly one stamp is put on the envelope. It is guaranteed that the given stamps correspond to some valid route from some city to some other city.
|
|
|
|
Output
|
|
|
|
Output n + 1 numbers — indexes of cities in one of the two possible routes of the letter.
|
|
|
|
Examples
|
|
|
|
Input
|
|
|
|
2
|
|
1 100
|
|
100 2
|
|
|
|
|
|
Output
|
|
|
|
2 100 1
|
|
|
|
Input
|
|
|
|
3
|
|
3 1
|
|
100 2
|
|
3 2
|
|
|
|
|
|
Output
|
|
|
|
100 2 3 1
|
|
|
|
Here is a reference code to solve this task. You can use this to help you genereate cases or validate the solution.
|
|
```python
|
|
from math import *
|
|
from Queue import *
|
|
|
|
|
|
n = int(raw_input())
|
|
nbr = dict()
|
|
cand = set()
|
|
for i in range(n):
|
|
l = raw_input().split()
|
|
if l[0] not in nbr:
|
|
nbr[l[0]] = [l[1]]
|
|
cand.add(l[0])
|
|
else:
|
|
nbr[l[0]].append(l[1])
|
|
cand.remove(l[0])
|
|
if l[1] not in nbr:
|
|
nbr[l[1]] = [l[0]]
|
|
cand.add(l[1])
|
|
else:
|
|
nbr[l[1]].append(l[0])
|
|
cand.remove(l[1])
|
|
for v in cand:
|
|
break
|
|
marked = set()
|
|
marked.add(v)
|
|
res = [v]
|
|
Q = [v]
|
|
while len(Q) > 0:
|
|
v = Q.pop()
|
|
for i in nbr[v]:
|
|
if i not in marked:
|
|
marked.add(i)
|
|
res.append(i)
|
|
Q.append(i)
|
|
print(\" \".join(res))
|
|
```
|
|
|
|
|
|
请完成上述谜题的训练场环境类实现,包括所有必要的方法。
|
|
"""
|
|
|
|
from bootcamp import Basebootcamp
|
|
import random
|
|
import re
|
|
from bootcamp import Basebootcamp
|
|
|
|
class Cmailstampsbootcamp(Basebootcamp):
|
|
def __init__(self, min_n=1, max_n=10):
|
|
self.min_n = max(1, min_n)
|
|
self.max_n = min(max_n, 10**5) # Enforce problem constraint
|
|
|
|
def case_generator(self):
|
|
n = random.randint(self.min_n, self.max_n)
|
|
# Generate unique non-consecutive city IDs
|
|
cities = set()
|
|
while len(cities) < n + 1:
|
|
new_city = random.randint(1, 10**9)
|
|
cities.add(new_city)
|
|
cities = list(cities)
|
|
# Create valid path through all cities
|
|
path = random.sample(cities, k=n+1)
|
|
# Generate stamps with random direction
|
|
stamps = []
|
|
for i in range(n):
|
|
a, b = path[i], path[i+1]
|
|
if random.choice([True, False]):
|
|
stamps.append([a, b])
|
|
else:
|
|
stamps.append([b, a])
|
|
random.shuffle(stamps) # Ensure stamp order randomization
|
|
return {
|
|
'n': n,
|
|
'stamps': stamps,
|
|
'correct_path': path
|
|
}
|
|
|
|
@staticmethod
|
|
def prompt_func(question_case) -> str:
|
|
stamps = "\n".join(f"{a} {b}" for a, b in question_case['stamps'])
|
|
return (
|
|
"Reconstruct the letter's route from postal stamps.\n"
|
|
"Rules:\n"
|
|
"1. Cmailstamps must be a single continuous path\n"
|
|
"2. Each city visited exactly once\n"
|
|
f"Stamps ({question_case['n']}):\n{stamps}\n"
|
|
f"Output format: [answer]{' X'*(question_case['n'])}[/answer] "
|
|
"(replace X with numbers)\n"
|
|
"Example: [answer]100 2 3 1[/answer]"
|
|
)
|
|
|
|
@staticmethod
|
|
def extract_output(output):
|
|
matches = re.findall(r'\[answer\]\s*((?:\d+\s*)+)\[/answer\]', output)
|
|
if not matches:
|
|
return None
|
|
try:
|
|
return list(map(int, re.split(r'\s+', matches[-1].strip())))
|
|
except (ValueError, AttributeError):
|
|
return None
|
|
|
|
@classmethod
|
|
def _verify_correction(cls, solution, identity):
|
|
expected = identity['correct_path']
|
|
return solution in [expected, expected[::-1]]
|