Unrelated to AfD, which would still win districts in the East, yes. I'd actually prefer a two-round system with direct popular vote similar to France rather than proportional representation. It just makes governing easier and strengthens MPs bound to their district.
More in general, I'd also prefer switching to a presidential system with the head of state and head of government being a single office holder elected by popular vote. I'd also change the Bundesrat (upper house, whose members are just state cabinet officers or appointed by state governments). Either reform the upper chamber or even abolish it entirely. I guess I'm in a minority with these opinions though.
Linke is probably dead anyway.
Are you looking for a German Charles de Gaulle?