Solução diferente a do professor, qual a melhor forma?

<meta charset="UTF-8">

<canvas width="600" height="400"></canvas>

<script>

    function desenhaQuadrado(x, y, tamanho, cor) {

        pincel.fillStyle = cor;
        pincel.fillRect(x, y, tamanho, tamanho)
        pincel.fill();
    }

    function desenhaCirculo(x, y, raio, cor) {

        pincel.fillStyle = cor;
        pincel.beginPath();
        pincel.arc(x, y, raio, 0, 2 * 3.14);
        pincel.fill();

    }

    function desenhaPaletaDeCores() {

        desenhaQuadrado(xVermelho, yQuadrados, tamanhoQuadrados, 'red');
        desenhaQuadrado(xVerde, yQuadrados, tamanhoQuadrados, 'green');
        desenhaQuadrado(xAzul, yQuadrados, tamanhoQuadrados, 'blue');

    }

    function lidaComMovimentoDoMouse(evento) {

        var x = evento.pageX - tela.offsetLeft;
        var y = evento.pageY - tela.offsetTop;

        if (desenha 
            && x >= 0
                && x <= 50
                    && y >= 0
                        && y <= 50) {
            corAtual = cores[0];
        } else if (desenha 
            && x >= 50
                && x <= 100
                    && y >= 0
                        && y <= 50) {
            corAtual = cores[1];
        } else if (desenha 
            && x >= 100
                && x <= 150
                    && y >= 0
                        && y <= 50) {
            corAtual = cores[2];
        } else if(desenha) {

            desenhaCirculo(x, y, 5, corAtual);
        }
    }

    function habilitaDesenhar() {

        desenha = true;
    }

    function desabilitaDesenhar() {

        desenha = false;
    }

    var tela = document.querySelector('canvas');
    var pincel = tela.getContext('2d');
    pincel.fillStyle = 'lightgray';
    pincel.fillRect(0, 0, 600, 400);

    var desenha = false;
    var cores =  ['red', 'green', 'blue'];
    var corAtual = cores[0];
    var xVermelho = 0;
    var xVerde  = 50;
    var xAzul = 100;
    var yQuadrados = 0;
    var tamanhoQuadrados = 50;

    desenhaPaletaDeCores(); // mostra os quadrados de seleção de cores

    tela.onmousemove = lidaComMovimentoDoMouse;

    tela.onmousedown = habilitaDesenhar;

    tela.onmouseup = desabilitaDesenhar;

</script>

Bom dia,

Eu fiz dessa maneira, porque achei que teria menos escrita, entretanto encontrei os seguintes problemas:

1. Quando eu coloco o o array de cores na variável corAtual, ele não lê as cores, mesmo que eu coloque corAtual[0] ou corAtual[1] ou corAtual[2];
2. Pelo fato de não ter colocado onclick, as cores só alternam quando se pressiona o botão do mouse em movimento sobre os quadrados de cor.

Gostaria de saber se a solução do professor é mais indicada, pois cria funções e variáveis próprias ou se soluções mais enxutas são preferíveis.

`