hola Iván, lo que proponés es correcto. Resulta una alternativa muy análoga a la solución propuesta, solamente que recorrés por columna en lugar de por fila. La representación de s y t deberían ajustarse a tu propuesta para que el tiempo de ejecución no se dispare.
saludos
saludos