Shift which problems to where? Legalizing will certainly substantially reduce the problem of drug-war-inflated profits going to criminals.
Stop making sense.
Criminals don’t ‘go straight’ when a criminal enterprise is eliminated.
They expand/transform to another crime. Such as other drugs, extortion of the legal growers, human trafficking, kidnapping, etc.
Not to mention the problem of providing drugs to humans, and the effect on the person and society. Addicts commit crime to purchase their drugs, or abandon/ignore their children, and so on...