[i][(i + 1) % 3] を [i * N + (i + 1) % N] にするだけでしょ。