It really is a crapshoot, TBH. I would say that phones in the Apple store are probably on iOS 5.1, due to high turnover. But, you may find one in a store somewhere that still has the older version.
It really is a game of chance, with the odds decreasing against finding one with iOS 5.0.1 as more time goes by.
Marilyn