星新一の作品にあったね。犯罪者はどんな罪だろうと火星送りになって、砂漠のような火星で、ボタンを押すと水が出る機械を渡されるけど、罪の重さによって水の出る回数が違うみたいなやつ。