[[練習問題]]
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 # main.py
 from collections import namedtuple
 
 X = namedtuple('X', 'letter, pos')
 Word = namedtuple('Word', 'letter, pos')
 
 class Solver:
     def __init__(self, s):
         self.copy = s
         self.arr = list(s)
         self.cost = 0
 
     def __change(self, letter, cost, *pos):
         for p in pos:
             self.arr[p] = letter
         self.cost += cost * len(pos)
 
     __to_N = lambda self, *pos: self.__change('N', 10, *pos)
     __to_O = lambda self, *pos: self.__change('O', 15, *pos)
     __change_to_N = lambda self, *pos: self.__change('N', 10, *pos)
     __change_to_O = lambda self, *pos: self.__change('O', 15, *pos)
 
     def to_palindromically(self):
         sz = len(self.arr)
         for i in xrange(sz/2):
             j = -1 - i
             if self.arr[i] == self.arr[j]:
                 if self.arr[i] == '*':
                     self.__to_N(i, j)
                     self.__change_to_N(i, j)
             else:
                 cnt = (self.arr[i], self.arr[j]).count('*')
                 if cnt == 0:
                 if not '*' in (self.arr[i], self.arr[j]):
                     return (self.copy, -1)
                 elif cnt == 1:
                     srt = sorted((X(self.arr[i], i), X(self.arr[j], j)))
                     if srt[-1].letter == 'N':
                         self.__to_N(srt[0].pos)
                     elif srt[-1].letter == 'O':
                         self.__to_O(srt[0].pos)
                 srt = sorted((Word(self.arr[i], i), Word(self.arr[j], j)))
                 if srt[-1].letter == 'N':
                     self.__change_to_N(srt[0].pos)
                 elif srt[-1].letter == 'O':
                     self.__change_to_O(srt[0].pos)
         if sz & 1 and self.arr[sz/2] == '*':
             self.__to_N(sz/2)
         return (''.join(self.arr), self.cost)
             self.__change_to_N(sz/2)
         return ''.join(self.arr), self.cost


 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 # test.py
 from main import Solver
 from collections import namedtuple
 import unittest
 
 T = namedtuple('T', 'input_data, expected')
 
 test_cases = [
   T('NOO*', ('NOON', 10)),
   T('NO*N', ('NOON', 15)),
   T('NO',   ('NO', -1)),
   T('O*O',  ('ONO', 10)),
   T('*********',  ('NNNNNNNNN', 90)),
   T('**********', ('NNNNNNNNNN', 100)),
   T('*O*******',  ('NONNNNNON', 85)),
   T('*O********', ('NONNNNNNON', 95)),
   T('*N*******',  ('NNNNNNNNN', 80)),
   T('*N********', ('NNNNNNNNNN', 90)),
   T('*N*****O*',  ('*N*****O*', -1)),
   T('*N******O*', ('*N******O*', -1)),
   T('****N****',  ('NNNNNNNNN', 80)),
   T('****N*****', ('NNNNNNNNNN', 90)),
   T('****O****',  ('NNNNONNNN', 80)),
   T('****NO****', ('****NO****', -1)),
 ]
 
 class TestSequence(unittest.TestCase):
     pass
 
 def test_generator(foobar, expected):
 def test_generator(input_data, expected):
     def test(self):
         obj = Solver(foobar)
         obj = Solver(input_data)
         self.assertEqual(obj.to_palindromically(), expected)
     return test
 
 for case in test_cases:
     name = 'test{0}'.format(case.input_data)
     test = test_generator(case.input_data, case.expected)
     setattr(TestSequence, name, test)
 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequence)
 unittest.TextTestRunner().run(suite)


トップ   新規 一覧 検索 最終更新   ヘルプ   最終更新のRSS