西电软院算法上机1的原题以及python代码实现(用python就不用怕查重啦)