Problem p03264 (python)

Fixed submission:
# abc108_a.py
# https://atcoder.jp/contests/abc108/tasks/abc108_a

# A - Pair /
# 実行時間制限: 2 sec / メモリ制限: 1024 MB
# 配点 : 100点

# 問題文
# 1以上 K以下の正の整数から、偶数と奇数ひとつずつの組を選ぶ方法の個数を求めてください。なお、選ぶ順番は考慮しません。

# 制約
#     2≤K≤100
#     Kは整数である

# 入力
# 入力は以下の形式で標準入力から与えられる。
# K

# 出力
# 1以上 K以下の正の整数から、偶数と奇数ひとつずつの組を選ぶ方法の個数を出力せよ。

# 入力例 1
# 3

# 出力例 1
# 2

# (2,1)と (2,3)が条件を満たします。

# 入力例 2
# 6

# 出力例 2
# 9

# 入力例 3
# 11

# 出力例 3
# 30

# 入力例 4
# 50

# 出力例 4
# 625


def calculation(lines):
    # N, K = list(map(int, lines[0].split()))
    N = int(lines[0])
    if N%2 == 0:
        ans = (N/2)**2
    else:
        ans = ((N+1)/2 * (N-1)/2)
    return [int(ans)]


# 引数を取得
def get_input_lines(lines_count):
    lines = list()
    for _ in range(lines_count):
        lines.append(input())
    return lines


# テストデータ
def get_testdata(pattern):
    if pattern == 1:
        lines_input = ['3']
        lines_export = [2]
    if pattern == 2:
        lines_input = ['6']
        lines_export = [9]
    if pattern == 3:
        lines_input = ['11']
        lines_export = [30]
    if pattern == 4:
        lines_input = ['50']
        lines_export = [625]
    return lines_input, lines_export


# 動作モード判別
def get_mode():
    import sys
    args = sys.argv
    if len(args) == 1:
        mode = 0
    else:
        mode = int(args[1])
    return mode


# 主処理
def main():
    mode = get_mode()
    if mode == 0:
        lines_input = get_input_lines(1)
    else:
        lines_input, lines_export = get_testdata(mode)

    lines_result = calculation(lines_input)

    for line_result in lines_result:
        print(line_result)

    # if mode > 0:
    #     print(f'lines_input=[{lines_input}]')
    #     print(f'lines_export=[{lines_export}]')
    #     print(f'lines_result=[{lines_result}]')
    #     if lines_result == lines_export:
    #         print('OK')
    #     else:
    #         print('NG')


# 起動処理
if __name__ == '__main__':
    main()

Buggy submission (verdict: Runtime Error):
# abc108_a.py
# https://atcoder.jp/contests/abc108/tasks/abc108_a

# A - Pair /
# 実行時間制限: 2 sec / メモリ制限: 1024 MB
# 配点 : 100点

# 問題文
# 1以上 K以下の正の整数から、偶数と奇数ひとつずつの組を選ぶ方法の個数を求めてください。なお、選ぶ順番は考慮しません。

# 制約
#     2≤K≤100
#     Kは整数である

# 入力
# 入力は以下の形式で標準入力から与えられる。
# K

# 出力
# 1以上 K以下の正の整数から、偶数と奇数ひとつずつの組を選ぶ方法の個数を出力せよ。

# 入力例 1
# 3

# 出力例 1
# 2

# (2,1)と (2,3)が条件を満たします。

# 入力例 2
# 6

# 出力例 2
# 9

# 入力例 3
# 11

# 出力例 3
# 30

# 入力例 4
# 50

# 出力例 4
# 625


def calculation(lines):
    # N, K = list(map(int, lines[0].split()))
    N = int(lines[0])
    if N%2 == 0:
        ans = (N/2)**2
    else:
        ans = ((N+1)/2 * (N-1)/2)
    return [int(ans)]


# 引数を取得
def get_input_lines(lines_count):
    lines = list()
    for _ in range(lines_count):
        lines.append(input())
    return lines


# テストデータ
def get_testdata(pattern):
    if pattern == 1:
        lines_input = ['3']
        lines_export = [2]
    if pattern == 2:
        lines_input = ['6']
        lines_export = [9]
    if pattern == 3:
        lines_input = ['11']
        lines_export = [30]
    if pattern == 4:
        lines_input = ['50']
        lines_export = [625]
    return lines_input, lines_export


# 動作モード判別
def get_mode():
    import sys
    args = sys.argv
    if len(args) == 1:
        mode = 0
    else:
        mode = int(args[1])
    return mode


# 主処理
def main():
    mode = get_mode()
    if mode == 0:
        lines_input = get_input_lines(2)
    else:
        lines_input, lines_export = get_testdata(mode)

    lines_result = calculation(lines_input)

    for line_result in lines_result:
        print(line_result)

    # if mode > 0:
    #     print(f'lines_input=[{lines_input}]')
    #     print(f'lines_export=[{lines_export}]')
    #     print(f'lines_result=[{lines_result}]')
    #     if lines_result == lines_export:
    #         print('OK')
    #     else:
    #         print('NG')


# 起動処理
if __name__ == '__main__':
    main()