概要

\(N\) 個のビルの座標 \(X_1, X_2, \dots, X_N\) が与えられるので、\(K\) 番目のビルの座標 \(X_K\) を特定し、そこから \(1\) を引いた値（\(X_K - 1\)）を求める問題です。

考察

この問題で重要な点は、「どの情報が必要で、どのような計算を行うか」を正確に把握することです。

• 必要な情報: ビルの総数 \(N\)、目印となるビルの番号 \(K\)、そして \(K\) 番目のビルの座標 \(X_K\) です。
• 計算内容: \(X_K - 1\) を計算します。
• 注意点:
  • 座標 \(X_i\) は最大で \(10^{18}\) という非常に大きな値になります。プログラミング言語によっては 64ビット整数型（long long など）を使う必要がありますが、Pythonでは標準の整数型でこの大きさをそのまま扱えます。
  • \(N\) が最大 \(10^5\) と多いため、入力を 1 つずつ読み込むよりも、まとめて読み込む方が実行時間を短縮できる場合があります。

アルゴリズム

1. 入力から \(N\) と \(K\) を受け取ります。
2. 続いて与えられる \(N\) 個の座標の中から、\(K\) 番目の値 \(X_K\) を取得します。
   • 多くのプログラミング言語の配列（リスト）は 0 から数え始めるため、問題文の「\(K\) 番目」はインデックス K-1 に相当することに注意してください。
3. \(X_K - 1\) を計算し、その結果を出力します。

計算量

• 時間計算量: \(O(N)\)
  • \(N\) 個の入力を読み込む処理が支配的です。
• 空間計算量: \(O(N)\)
  • すべての入力をリストに格納する場合、入力の数に比例したメモリを使用します。

実装のポイント

• 高速な入力: sys.stdin.read().split() を使用することで、すべての入力データを一括で取得し、空白や改行で分割したリストとして扱うことができます。これにより、大量の入力データがある場合でも効率的に処理できます。

• インデックスの指定: 提供されたコードでは、input_data の 0 番目が \(N\)、1 番目が \(K\)、2 番目以降が \(X_1, X_2, \dots\) となっています。そのため、\(X_K\) は input_data[1 + K] で直接アクセスすることが可能です。

  ソースコード

import sys

def main():
    # 標準入力からすべてのデータを取得し、空白や改行で分割します
    input_data = sys.stdin.read().split()
    
    if not input_data:
        return

    # 1つ目の要素がN（ビルの総数）、2つ目の要素がK（目印のビル番号）です
    # N = int(input_data[0])
    K = int(input_data[1])
    
    # 各ビルの座標 X_i は input_data[2] 以降に格納されています
    # K番目のビルの座標 X_K は、インデックス 1 + K の位置にあります
    x_k = int(input_data[1 + K])
    
    # 待ち合わせ場所の座標 (X_K - 1) を出力します
    print(x_k - 1)

if __name__ == "__main__":
    main()

この解説は gemini-3-flash-thinking によって生成されました。