InternBootcamp/internbootcamp/bootcamp/cmailstamps/cmailstamps.py
2025-05-23 15:27:15 +08:00

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]]