Problem:
品質検査

問題

あなたはある機械の製造工場で品質管理の責任者をしている． この機械には， 部品として電源とモーターとケーブルが必要である． 製造工場には電源が a 個， モーターが b 個， ケーブルが c 個あり， それぞれ 1 から a まで， a+1 から a+b まで， a+b+1 から a+b+c までの番号が付いている． 困ったことに， 部品の中に故障しているものがあるかもしれない． 工場ではどの部品が故障していてどの部品が正常であるかを知りたい．

そこで， 工場では次の方法で部品を検査した． 電源とモーターとケーブルを１つずつ持ってきてつなぎ， 動作させてみる． このとき， ３つの部品がすべて正常であるときは正しく動作して「合格」とわかる． ３つの中に故障している部品が１つでも入っているときは正しく動作しないので「不合格」とわかる． （工場で作っている機械はとても精密なので， 故障した部品がまざっているのに偶然正しく動作してしまうなどということは起きないのだ．）

あなたには検査結果のリストが渡される． 検査結果のリストの各行には， 検査に使った電源とモーターとケーブルの番号と， 検査が合格だったか不合格だったかが書かれている．

検査結果のリストが与えられたとき， すべての部品を， 検査結果から確実に故障しているとわかる部品と， 確実に正常とわかる部品と， 検査結果からは故障しているとも正常であるとも決まらない部品に分類するプログラムを作成せよ．

入力

入力は複数のデータセットからなる．各データセットの形式は以下の通りである．入力は３つのゼロを含む行で終了する．

1 行目には 3 個の整数が空白区切りで書かれており， 順に電源の個数 a, モーターの個数 b, ケーブルの個数 c を表す．

2 行目には 1 個の整数が書かれており， 検査結果のリストに含まれる検査の回数 N が書かれている．

続く N 行は検査結果のリストを表す． 各行には， 4 個の整数 i, j, k, r が１つの空白を区切りとして書かれており， 電源 i とモーター j とケーブル k をつないで検査した結果が， 「合格」 (r=1 のとき) か「不合格」 (r=0 のとき) だったことを表す.

a, b, c, N は 1 ≤ a, b, c ≤ 100, 1 ≤ N ≤ 1000 を満たす.

データセットの数は 5 を超えない．

出力

データセットごとに以下の形式で出力する．各データセットの出力は a+b+c 行からなる.

i 行目 (1 ≤ i ≤ a+b+c)：

検査結果から部品 i が故障しているとわかる場合は 0 を出力する.

検査結果から部品 i が正常とわかる場合は 1 を出力する.

検査結果から部品 i が故障しているか正常であるかが決まらない場合は 2 を出力する.

入出力例

入力例

2 2 2
4
2 4 5 0
2 3 6 0
1 4 5 0
2 3 5 1
0 0 0

出力例

2
1
1
0
1
0

上記問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in Haskell (judge verdict: Accepted)

import Control.Applicative
import Control.Monad
import Data.List
import Data.IntSet (IntSet)
import qualified Data.IntSet as IS

main :: IO ()
main = do
  [a, b, c] <- f
  unless (a == 0 && b == 0 && c == 0) $ do
    n <- readLn
    solve (a+b+c) <$> replicateM n f >>= mapM_ print
    main
  where
    f = map read <$> words <$> getLine

solve :: Int -> [[Int]] -> [Int]
solve abc xss = map f [1 .. abc]
  where
    rms = foldl' (\s [i, j, k, r] -> if r == 0
                                     then s
                                     else IS.insert i (IS.insert j (IS.insert k s))) IS.empty xss
    bms = foldl' (\s [i, j, k, r] -> case (IS.member i rms, IS.member j rms, IS.member k rms) of
                                      (True, True, False) -> IS.insert k s
                                      (True, False, True) -> IS.insert j s
                                      (False, True, True) -> IS.insert i s
                                      _ -> s
                   ) IS.empty xss
    f x = if IS.member x bms
          then 0
          else if IS.member x rms
               then 1
               else 2