結果

問題 No.118 門松列(2)
ユーザー ducktailducktail
提出日時 2018-06-18 21:08:00
言語 Haskell
(9.8.2)
結果
AC  
実行時間 145 ms / 5,000 ms
コード長 836 bytes
コンパイル時間 4,883 ms
コンパイル使用メモリ 180,832 KB
実行使用メモリ 17,152 KB
最終ジャッジ日時 2024-06-30 17:04:32
合計ジャッジ時間 6,161 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
6,812 KB
testcase_01 AC 138 ms
17,152 KB
testcase_02 AC 133 ms
17,024 KB
testcase_03 AC 137 ms
17,024 KB
testcase_04 AC 137 ms
17,024 KB
testcase_05 AC 145 ms
17,024 KB
testcase_06 AC 1 ms
6,944 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,948 KB
testcase_09 AC 126 ms
16,128 KB
testcase_10 AC 19 ms
9,472 KB
testcase_11 AC 26 ms
9,984 KB
testcase_12 AC 14 ms
8,832 KB
testcase_13 AC 10 ms
8,448 KB
testcase_14 AC 124 ms
16,000 KB
testcase_15 AC 4 ms
6,940 KB
testcase_16 AC 59 ms
12,800 KB
testcase_17 AC 9 ms
8,192 KB
testcase_18 AC 27 ms
9,984 KB
testcase_19 AC 49 ms
11,776 KB
testcase_20 AC 53 ms
11,904 KB
testcase_21 AC 98 ms
15,104 KB
testcase_22 AC 21 ms
9,472 KB
testcase_23 AC 84 ms
13,952 KB
testcase_24 AC 29 ms
9,856 KB
testcase_25 AC 44 ms
11,264 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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.Applicative ((<$>), (<*>))
import Data.ByteString.Char8 (ByteString)
import qualified Data.ByteString.Char8 as B
import Data.List (unfoldr)
import Data.List (sort, group)
import Data.Char (isSpace)

main :: IO ()
main = solve <$> readLn <*> (readil B.readInt <$> B.getLine) >>= print

solve :: Int -> [Int] -> Int
solve n xs = flip mod (10^9+7) $ comb n 3 - (sum . map (flip comb 3) $ l3) - (sum . map (\m -> (n-m) * comb m 2) $ l2)
  where l2 = filter (>=2) . map length . group . sort $ xs
        l3 = filter (>=3) l2

comb n m | m == 0 = 1
         | n < 2 * m = comb n (n-m)
         | otherwise = comb (n-1) (m-1) * n `div` m

readil :: Integral a =>  (ByteString -> Maybe (a, ByteString)) -> ByteString -> [a]
readil f = unfoldr g
  where
    g s = do
      (n, s') <- f s
      return (n, B.dropWhile isSpace s')
0