どの言語でも躓く、競技プログラミング対策についての質問です。

Question

import Control.Applicative
main = do
  ii <- fmap (fmap (read :: String -> Int) . words) . lines <$> getContents
    break  ( ii == ("\n","\n")) 
  print $ ii

複数行の入力を受け取って、
改行を二回入れたらブレイクしたいなって思い、
このようなコードを書いてみましたが
やっぱりそうは問屋が卸さないようです。

huku.hs:3:65: error:
    • Couldn't match expected type ‘((a0 -> Bool)
                                     -> [a0] -> ([a0], [a0]))
                                    -> Bool -> IO String’
                  with actual type ‘IO String’
    • The function ‘getContents’ is applied to two arguments,
      but its type ‘IO String’ has none
      In the second argument of ‘(<$>)’, namely
        ‘getContents break (ii == ("\n", "\n"))’
      In a stmt of a 'do' block:
        ii <- fmap (fmap (read :: String -> Int) . words) . lines
              <$> getContents break (ii == ("\n", "\n"))

huku.hs:4:14: error:
    • Variable not in scope: ii :: ([Char], [Char])
    • Perhaps you meant one of these:
        ‘id’ (imported from Prelude), ‘pi’ (imported from Prelude)

このように、親切ではありますが
たくさんのエラーが出てしまい
参ってしまいました。
助けてください。お願いします。

ありがとうございます。

改行二回のイメージは、二回連続でエンターを押した時です。
もう入力する値がない場合に、値を決定して
入力から抜けたいです。

Comments

[ozwk]

「改行を二回入れたらブレイクしたい」の具体例もらえますか?

[raccy]

「空白行が二回続く」でしょうか、それとも、「改行が二回続く(実質空白行は一回のみ)」でしょうか。後者だと一応できたのですが…。

[Bolshikh_i_mal]

ありがとうございます。俺がイメージしていたのは後者の方です。

[Bolshikh_i_mal]

ちなみに、ですが、ここで入力したリストを一つのリストとする場合 fordを使えば出来ますか？

Answer 1

改行が二回、つまり、空白行が一つあればと言うことであれば、

43 3
2 2
63
324 8 9
3 4
98

こんな感じの入力について(わかりにくいですが最後に空白行ありです)、全ての数字を足す場合は、

main :: IO ()
main = do
    list <- takeWhile (not . null) . map (map read . words) . lines <$> getContents
    print $ sum $ map sum list

という風になるかと思います。

Comments

[Bolshikh_i_mal]

最高です！

Answer 2

こういうことでしょうか？

import Control.Applicative
import Data.List (tails, isPrefixOf)

main = do
  ls <- map (map (read :: String -> Integer) . words . head) . takeWhile (not . isPrefixOf ["", ""]) . tails . lines <$> getContents
  print ls

コツとしてはtails . linesとすることで1行ずつずれたリストを用意しておき、このリストの先頭が["", ""]となるところまで、つまり改行のみの行が2回連続で続くところまでをtakeWhileで取得しています。

Comments

[Bolshikh_i_mal]

ありがとうございます。感謝感激です。