結果

問題 No.118 門松列(2)
ユーザー ducktailducktail
提出日時 2018-06-18 21:08:00
言語 Haskell
(9.8.2)
結果
AC  
実行時間 153 ms / 5,000 ms
コード長 836 bytes
コンパイル時間 1,343 ms
コンパイル使用メモリ 163,384 KB
実行使用メモリ 21,176 KB
最終ジャッジ日時 2023-09-13 06:56:22
合計ジャッジ時間 4,329 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
9,628 KB
testcase_01 AC 152 ms
21,036 KB
testcase_02 AC 142 ms
20,148 KB
testcase_03 AC 153 ms
20,180 KB
testcase_04 AC 152 ms
20,120 KB
testcase_05 AC 153 ms
21,176 KB
testcase_06 AC 3 ms
7,508 KB
testcase_07 AC 3 ms
7,556 KB
testcase_08 AC 3 ms
7,688 KB
testcase_09 AC 132 ms
20,028 KB
testcase_10 AC 22 ms
12,868 KB
testcase_11 AC 32 ms
13,584 KB
testcase_12 AC 22 ms
12,556 KB
testcase_13 AC 12 ms
11,880 KB
testcase_14 AC 132 ms
20,044 KB
testcase_15 AC 5 ms
9,440 KB
testcase_16 AC 63 ms
16,196 KB
testcase_17 AC 12 ms
11,788 KB
testcase_18 AC 32 ms
13,644 KB
testcase_19 AC 52 ms
15,648 KB
testcase_20 AC 62 ms
15,236 KB
testcase_21 AC 102 ms
17,984 KB
testcase_22 AC 22 ms
13,028 KB
testcase_23 AC 92 ms
17,040 KB
testcase_24 AC 32 ms
13,660 KB
testcase_25 AC 52 ms
15,684 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.6.1/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