用python实现的时间片轮转算法

用python实现的时间片轮转算法，在代码优化上还需要提升

import random

def createP(p_num):
    """
    创建字典，根据输入的进程数进行创建进程字典，时间为随机1-9s之间
    """
    p_dict = {}
    # 获取一个a-z的列表
    alphabet_list = list(map(chr, range(ord('a'), ord('z') + 1)))
    
    for i in range(0, p_num):
        # 默认进程名为a-z顺序排序
        p_dict[alphabet_list[i]] = random.randint(1,9)
    return p_dict

def lunzhuanP(p_dict, p_num):
    """
    开始轮转
    """
    print("进程名称"+ " "*10 +"每个进程需要工作的时间")
    p_time = {}
    p_list = [[-1 for i in range(4)] for j in range(p_num+1)]
    # 添加输出头
    p_list[0][0] = "Name"
    p_list[0][1] = "run"
    p_list[0][2] = "req"
    p_list[0][3] = "status"
    z = 1
    for k, v in p_dict.items():
        # 将进程初始化设置初始值
        p_list[z][0] = k
        p_list[z][1] = 0
        p_list[z][2] = v
        p_list[z][3] = "R"
        p_time[k] = 0
        z += 1
    
    for k, v in p_dict.items():
        print(k + "          " ,v)
    j = 1
    t = 1
    while True:
        if len(p_list) == 1 :
            print("进程执行完毕")
            for k, v in p_time.items():
                print("进程%s的执行周期为：%s"%(k, v))
            break

        # 判断是否有进程执行完毕
        for y in range(1, len(p_list)-1):

            if p_list[y][3] == "E":
    
                del p_list[y]

        if j <= len(p_list) -1 :
            
            if p_list[j][1] +1 <= p_list[j][2]:

                p_list[j][1] += 1

                if p_list[j][1] == p_list[j][2]:

                    p_list[j][3] = "E"
            else:

                del p_list[j]
                continue

        else:
            j = j - len(p_list) + 1
            continue
        
        print("cpu时刻: ", t)
        print("正在执行的进程：", p_list[j][0])
        p_time[p_list[j][0]] += 1
        j += 1
        t += 1
        for i in range(len(p_list)):
            
            print(p_list[i])

def main():

    while True:
        p_num = input("请输入进程数：")
        if p_num.isdigit():
            p_num = int(p_num)
            p_dict = createP(p_num)
            lunzhuanP(p_dict, p_num)

        else:
            print("输入的不是有效数字")

if __name__ == "__main__":

    main()

运行截图


D9CA0058-031E-4DF8-9288-5A9CADDBCE27.png