結果

問題 No.75 回数の期待値の問題
コンテスト
ユーザー lam6er
提出日時 2025-03-20 21:19:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 134 ms / 5,000 ms
コード長 1,405 bytes
コンパイル時間 234 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 76,436 KB
最終ジャッジ日時 2025-03-20 21:20:36
合計ジャッジ時間 2,101 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve():
    import sys
    K = int(sys.stdin.readline())
    if K == 0:
        print(0.0)
        return
    
    size = K
    matrix = []
    for s in range(size):
        row = [0.0] * (size + 1)
        row[s] = 6.0  # Coefficient for dp[s]
        for d in range(1, 7):
            current_sum = s + d
            if current_sum == K:
                continue
            elif current_sum > K:
                row[0] -= 1.0
            else:
                x = current_sum
                row[x] -= 1.0
        row[size] = 6.0  # RHS is always 6
        matrix.append(row)
    
    n = len(matrix)
    for i in range(n):
        # Partial pivoting
        max_row = i
        for j in range(i, n):
            if abs(matrix[j][i]) > abs(matrix[max_row][i]):
                max_row = j
        matrix[i], matrix[max_row] = matrix[max_row], matrix[i]
        
        pivot = matrix[i][i]
        if pivot == 0:
            print(0.0)
            return
        
        # Normalize pivot row
        for j in range(i, n + 1):
            matrix[i][j] /= pivot
        
        # Eliminate other rows
        for j in range(n):
            if j != i:
                factor = matrix[j][i]
                for k in range(i, n + 1):
                    matrix[j][k] -= factor * matrix[i][k]
    
    solution = [row[n] for row in matrix]
    print("{0:.5f}".format(round(solution[0], 5)))

solve()
0