川端康成の小説

高校の時に模試で出題された川端康成の小説を探しています。

出題された部分は以下のような感じです。
別れた彼女か妻が、家が火事になっているのに逃げようとしない。どうしてなんだと聞くと、あなたが安全な場所にいて、そちらへ行くとあなたの所へいくことになる。あなたの方へ行きたくないから、そちらへは行かないと答える夢をみる。目が覚めたあと主人公が、別れた女に未練があり戻ってきて欲しいと思っているが、自分の潜在的な部分では、もう彼女が戻ってこないとあきらめているんだなと気付く。

模試で読んだ時に、本の題名を覚えておいて後で読もうと思いながらウン十年たってしまいました（笑）すっかり題名を忘れてしまったので、ぜひ小町の皆様にお知恵を貸していただけないでしょうか。よろしくお願いします。