ABC081A - Placing Marbles

题意

输入一个长度为 3 的字符串，每个字符为 0 或 1。统计其中 1 的个数并输出。

例如 101 中有 2 个 1，输出 2。

思路

本题不涉及算法优化，重点是练习 Haskell 的基础 IO 和字符串操作。

核心流程：

用 getLine 读入一行字符串
用 filter (== '1') 过滤出所有 '1' 字符
用 length 统计个数
用 print 输出结果

代码

haskell

1
2
3
4
5
6
7
8
9
10
11
12{-
 Author by Rainboy blog: https://rainboylv.com github: https://github.com/rainboylvx
 rbook: -> https://rbook.roj.ac.cn  https://rbook2.roj.ac.cn
 rainboy的学习导航网站: https://idx.roj.ac.cn
 create_at: 2026-07-09 20:47
 update_at: 2026-07-09 20:47
-}
main :: IO ()
main = do
    s <- getLine
    let cnt = length (filter (== '1') s)
    print cnt

另一种写法，使用 dropWhile 去掉前导 0 后计数：

haskell

1
2
3
4
5
6
7
8
9
10
11
12{-
 Author by Rainboy blog: https://rainboylv.com github: https://github.com/rainboylvx
 rbook: -> https://rbook.roj.ac.cn  https://rbook2.roj.ac.cn
 rainboy的学习导航网站: https://idx.roj.ac.cn
 create_at: 2026-07-09 20:47
 update_at: 2026-07-09 20:47
-}
main :: IO ()
main = do
    s <- getLine
    let cnt = length . dropWhile (== '0') $ s
    print cnt

复杂度

时间复杂度 $O(1)$ ，空间复杂度 $O(1)$ 。

总结

Haskell 中处理简单输入输出和字符串过滤的基本模式： getLine 读入，filter 筛选，length 计数，print 输出。

ABC081A - Placing Marbles

目录

题意

思路

代码

复杂度

总结