結果

問題 No.22 括弧の対応
ユーザー e7a894e7a894
提出日時 2020-09-18 14:32:39
言語 Haskell
(9.10.1)
結果
AC  
実行時間 3 ms / 5,000 ms
コード長 661 bytes
コンパイル時間 1,715 ms
コンパイル使用メモリ 169,600 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 07:52:29
合計ジャッジ時間 2,418 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

import Control.Monad
import Data.List

main = do
  [n, k] <- map read . words <$> getLine :: IO [Int]
  s <- getLine
  
  let x = s !! (k - 1)
  let s' = if x == '(' then drop k s else take (k - 1) s
  let result = if x == '(' then checkL s' 0 0 else checkR (reverse s') 0 0
  
  putStrLn . show $ if x == '(' then result + k else (length s') - result + 1
  
checkL (c:cs) count no
  | c == '(' = checkL cs (count + 1) (no + 1)
  | c == ')' && count > 0 = checkL cs (count - 1) (no + 1)
  | otherwise = no + 1

checkR (c:cs) count no
  | c == ')' = checkR cs (count + 1) (no + 1)
  | c == '(' && count > 0 = checkR cs (count - 1) (no + 1)
  | otherwise = no + 1
0