第120回【Python】クラスの継承

現在取り組んでいるのは、paiza ラーニング問題集「クラス・構造体メニュー」になります。

はじめに

猫とキャンプと野球観戦と AWS が大好きな旦那、LeoSaki です。モフモフしたい。

Python をゼロから勉強してみよう、のコーナー 120 回目です。

AWS は毎日のように利用していますが、なかなか新しい AWS についてのブログ記事が書けていません。 AWS のアップデートスピードは相変わらず早く、いろいろな変化に気づいていますし、それに対応するための勉強もしています。しかし、ブログ記事を書くことができるくらいまで調べられていない現状です。

それでは、今日も頑張ってみようと思います。

クラスの継承

paiza 国の大衆居酒屋で働きながらクラスの勉強をしていたあなたは、お客さんをクラスに見立てることで店内の情報を管理できることに気付きました。
全てのお客さんは、ソフトドリンクと食事を頼むことができます。
paiza 国の法律では、 20 歳以上のお客さんは成人とみなされ、お酒を頼むことができます。
20 歳未満のお客さんは未成年とみなされ、お酒を頼もうとした場合はその注文は取り消されます。
また、お酒を頼んだ場合、以降の全ての食事の注文 が毎回 200 円引きになります.

店内の全てのお客さんの数と注文の回数、各注文をしたお客さんの番号とその内容が与えられるので、各お客さんの会計を求めてください。

ヒント

注文について、20 歳未満のお客さんにできて、 20 歳以上のお客さんにできないことはないので、20歳未満のお客さんのクラスを作成して、それを継承して 20歳以上のお客さんのクラスを作成することで効率よく実装することができます。

N K
a_1
...
a_N
n_1 s_1 m_1
...
n_K s_K m_K

・ 1 行目では、お客さんの人数 N と注文の回数 K が与えられます。
・ 続く N 行のうち i 行目(1 ≦ i ≦ N)では、i 番目のお客さんの年齢が与えられます。
・ 続く K 行では、頼んだお客さんの番号 n_i , 注文の種類 s_i , 値段 m_i (1 ≦ i ≦ K) が与えられます。


すべてのテストケースにおいて、以下の条件をみたします。

・ 1 ≦ N , K ≦ 1000
・ 1 ≦ a_i ≦ 100 (1 ≦ i ≦ N)
・ 1 ≦ n_i ≦ N (1 ≦ i ≦ K)
・ 1 ≦ s_i ≦ N (1 ≦ i ≦ K) は “food" , “softdrink" , “alcohol" のいずれかです。
food , softdrink , alcohol はその注文が食事・ソフトドリンク・お酒であることを表しています。
・ 300 ≦ m_i ≦ 5000 (1 ≦ i ≦ K)


入力例

2 5
59
5
2 food 1223
1 alcohol 4461
1 alcohol 4573
1 alcohol 1438
2 softdrink 1581

出力例

10472
2804

クラスの継承。インヘリタンス。いろいろな問題の例題に、車クラスを継承してスポーツカークラスやトラッククラスを作りましょうというのがある。

Python
class Underage:
    def __init__(self):
        self.total = 0
        
    def alcohol(self,price):
        pass
    
    def softdrink(self,price):
        self.total += price
    
    def food(self,price):
        self.total += price
        
    def total_price(self):
        return self.total
        
class Overage(Underage):
    def __init__(self):
        super().__init__()
        self.flg = False
    
    def alcohol(self,price):
        self.flg = True
        self.total += price
        
    def food(self,price):
        if self.flg:
            self.total += price - 200
        else:
            self.total += price
        
N,K = map(int,input().split())
guests = [None] * N 
for i in range(N):
    a = int(input())
    if a < 20:
        guests[i] = Underage()
    else: 
        guests[i] = Overage()
        
for i in range(K):
    n,s,m = input().split()
    idx = int(n) - 1
    price = int(m)
    if s == 'food':
        guests[idx].food(price)
    elif s == 'alcohol':
        guests[idx].alcohol(price)
    elif s == 'softdrink':
        guests[idx].softdrink(price)

for guest in guests:
    print(guest.total_price())
VBA
## Underage
Private total As Long

Private Sub Class_Initialize()

    total = 0
    
End Sub

Property Let alcohol(price)

    
End Property

Property Let softdrink(price)

    total = total + price
    
End Property

Property Let food(price)

    total = total + price
    
End Property

Property Get totalPrice()

    totalPrice = total
    
End Property

## Overage
Implements Underage

Private total As Long
Private flg As Boolean

Private Sub Class_Initialize()

    total = 0
    flg = False
    
End Sub

Property Let Underage_food(price)

    If flg Then
        total = total + price - 200
    Else
        total = total + price
    End If
    
End Property

Property Let Underage_softdrink(price)

    total = total + price
    
End Property

Property Let Underage_alcohol(price)

    flg = True
    total = total + price
    
End Property

Property Get Underage_totalPrice()

    Underage_totalPrice = total
    
End Property

## 標準モジュール
Sub class_primer__inheritance()

    NK = Split(Cells(1, 1), " ")
    N = Val(NK(0))
    K = Val(NK(1))
    
    Dim cls() As New Underage
    ReDim cls(N - 1)
    
    For i = 0 To N - 1
        age = Cells(i + 2, 1)
        If age < 20 Then
            Set cls(i) = New Underage
        Else
            Set cls(i) = New Overage
        End If
    Next
    
    For i = 0 To K - 1
        S = Split(Cells(i + N + 2, 1), " ")
        idx = Val(S(0)) - 1
        ord = S(1)
        price = Val(S(2))
        If ord = "food" Then
            cls(idx).food = price
        ElseIf ord = "alcohol" Then
            cls(idx).alcohol = price
        ElseIf ord = "softdrink" Then
            cls(idx).softdrink = price
        End If
    Next
    
    For i = 0 To N - 1
        Debug.Print cls(i).totalPrice()
    Next
       
End Sub

最後に

VBA では継承がサポートされていないため、疑似継承という形で強引に継承させてみました。paiza が出してくれている境界値データを試してみて、間違いなく出力されることを確認しています。出来ちゃうものです。とてもとても面倒で、可読性は下がりますが。

それに比べて(比べるモノが違うというご指摘は重々承知しています)、Python の継承は分かりやすくて良いです。

引き続き、よろしくお願いいたします!

Pythonpaiza,学習,Python

Posted by LeoSaki