求一部关于数学推理的电影
讲的好像是几个数学家被约到一个房间里,那房间的四面墙后面有四个水泵,每隔好像四十分钟就会把墙往前推,需要数学家们解题来减缓水泵推墙的速度,要不他们就会被挤死.有一个片段是一个女数学家推理房间的灯亮的规律.求这部电影,谢谢.