I would think a non-sugary electrolyte replacement beverage like Nuun would be sufficient to offset electrolyte loss due to diarrhea.
I had wild fluctuations with my sodium and potassium immediately after the colon came out, and in fact remember vividly a nurse rushing into my room telling me I had low potassium and telling me I had to drink some liquid potassium. I remember it was blood red in color and had a vile test and I took like a guppy sip or two and the nurse practically tore my head off. She said I had to drink it NOW and I then did. I later read that at that time (early 1990s) there had been some cardiac arrests associated with giving potassium by IV in these low potassium situations, and this is why Mount Sinai made me drink this liquid potassium concoction that looked like blood. I know that you can't bombard the body suddenly with a high dose of potassium as by injection or it will cause a cardiac event, but drinking it should allow kidneys to do the job.
On that note however, I was hospitalized with a severe sodium overdose at my 30 days after takedown sodium party after eating anchovy pizza, Lays potato chips and V8 juice. They had put me on a high sodium diet, I took it to extreme. There was at 3 am or so the most severe pain in my side where my kidneys are, I was rushed to the ER, and the Doc said it was the highest sodium level he had ever seen and asked me what the hell I did and then told me to never do that again. I was young and quite stupid at the time. This was circa 1992.