Problem:
最長の階段

問題

1 から n までの整数がそれぞれ 1 つずつ書かれた n 枚のカードと,1 枚の白紙の
カードがある.これら n + 1 枚のカードの内,k 枚のカードが与えられる.ただし,
1 ≤ k ≤ n である.白紙のカードには 1 から n までの整数を 1 つ書くことができ
る.与えられたカードだけで,できるだけ長い連続した整数列を作りたい.

与えられるカードが入力されたときに, 与えられたカードから作ることができる連
続した整数列の最大長を出力するプログラムを作成せよ.

例1

n = 7, k = 5 とする.6, 2, 4, 7, 1 のカードが与えられたとき,このカードを使っ
て作れる連続した整数列のうち最長のものは 1, 2 であり,その長さは 2 である.

例2

n = 7, k = 5 とする.6, 2, 4, 7 と白紙のカードが与えられたとき,このカードを
使って作れる連続した整数列のうち最長のものは,白紙のカードに 5 を書くことに
よってできる 4, 5, 6, 7 であり,その長さは 4 である.

入力

入力は複数のデータセットからなる．各データセットは以下の形式で与えられる．入力はゼロ２つを含む行で終了する．

1 行目には,2 つの整数 n (1 ≤ n ≤ 100000) と k (1 ≤ k ≤ n) がこの順で 1 つの
空白を区切りとして書かれている.続く k 行には整数が 1 つずつ書かれており,与
えられる k 枚のカードに書かれている整数を表している.白紙のカードは 0 で表さ
れる.

採点用データのうち, 配点の 40% 分は 1 ≤ n ≤ 1000, 1 ≤ k ≤ 500 を, 配点の 20% 分は 1 ≤ n ≤ 60000, 1 ≤ k ≤ 50000 を. 配点の 40% 分は 1 ≤ n ≤ 100000, 1 ≤ k ≤ 100000 を満たす.

データセットの数は 5 を超えない．

出力

データセットごとに整数を1行に出力する．

入出力例

入力例

7 5
6
2
4
7
1
7 5
6
2
0
4
7
0 0

出力例

2
4

上記問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in Rust (judge verdict: Accepted)

use std::cmp;

fn main(){
  loop {
    let nk: Vec<usize> = read_vec();
    let n = nk[0];
    let k = nk[1];

    if n == 0 && k == 0 { break; }

    let mut cv: Vec<bool> = vec![false; n+1];

    for _ in 0 .. k {
      let c: usize = read();
      cv[c] = true;
    }

    let mut mxl: usize = 1;
    let mut av: Vec<usize> = vec![0; n+1];
    
    if cv[0] {
      for i in 1 .. n+1 {
        if cv[i] {
          if cv[i-1] {
            av[i] = av[i-1] + 1;
          } else {
            av[i] = 1;
          }
        }
      }
      
      let mut mv: Vec<usize> = vec![0; n+1];

      mv[n] = av[n];
      
      for i in (1 .. n).rev() {
        if av[i] > 0 {
          if av[i+1] > 0 {
            mv[i] = mv[i+1];
          } else {
            mv[i] = av[i];
          }
        }
      }

      for i in 1 .. n+1 {
        if !cv[i] {
          let s = 1 +
                  if i > 1 { mv[i-1] } else { 0 } +
                  if i < n { mv[i+1] } else { 0 };
          mxl = cmp::max(mxl, s);
        }
      }
    } else {
      for i in 1 .. n+1 {
        if cv[i] {
          if cv[i-1] {
            av[i] = av[i-1] + 1;
            mxl = cmp::max(mxl, av[i]);
          } else {
            av[i] = 1;
          }
        }
      }
    }
    
    println!("{}", mxl);
  }
}

fn read<T>() -> T
  where T: std::str::FromStr,
        T::Err: std::fmt::Debug
{
  let mut buf = String::new();
  std::io::stdin().read_line(&mut buf).expect("failed to read");
  buf.trim().parse().unwrap()
}

fn read_vec<T>() -> Vec<T>
  where T: std::str::FromStr,
        T::Err: std::fmt::Debug
{
  let mut buf = String::new();
  std::io::stdin().read_line(&mut buf).expect("failed to read");
  buf.split_whitespace().map(|e| e.parse().unwrap()).collect()
}