document.querySelector('button').onclick = function (e) {
e.target.style.background = 'red';
}
Или через атрибуты в HTML-тегах...
Чуть сложнее сделать наоборот, чтобы цвет остался после обновления. Придется использовать localStorage для хранения стиля.