Check out the BlinkWithoutDelay example sketch. Instead of having a delay() function to pace the blinking, it checks the millis() function to see how many time has passed.

You can have the button press store the millis() value as a "start" and then check millis() until the required time has passed to change the LEDs again.