将以下C++代码转换成python语言#include <stdio.h> #include <stdlib.h> #include <string.h> int cmp(const void *a,const void *b){ int *arr1 = *(int **)a; int *arr2 = *(int **)b; int wa = arr1[2]; int wb = arr2[2]; return wa - wb; } int compare(const void *a,const void *b){ return *(int *)a - *(int *)b; } int main(){ int i,j,num; scanf("%d",&num); // int arr[num][5]; int **arr = (int **)malloc(sizeof(int*)*num); for(i = 0;i < num;i++){ arr[i] = (int *)malloc(sizeof(int)*5); for(j = 0;j < 5;j++){ scanf("%d",&arr[i][j]); } } //按照y1对数组排序 qsort(arr,num,sizeof(int*),cmp); //判断是否属于基准灯同一行，若属于同一行设置为1，下次不再排序 int flag[num]; memset(flag,0x00,sizeof(int)*num); //收集结果 int res_arr[num],res_arr_cnt = 0; //灯大小一样，取第一个灯计算半径 int radius = (arr[0][3] - arr[0][1])/2; // printf("radius:%d",radius); for(i = 0;i < num - 1;i++){ if(flag[i] != 0){ continue; } //判断基准灯与略低于基准灯是否同一行 if(arr[i + 1][2] - arr[i][2] <= radius){ //属于同一行 if(arr[i][1] <= arr[i + 1][1]){ res_arr[res_arr_cnt++] = arr[i][0]; res_arr[res_arr_cnt++] = arr[i + 1][0]; }else{ res_arr[res_arr_cnt++] = arr[i + 1][0]; res_arr[res_arr_cnt++] = arr[i][0]; } flag[i] = 1; flag[i + 1] = 1; }else{ //不属于同一行 res_arr[res_arr_cnt++] = arr[i][0]; flag[i] = 1; } } //对输出列表排序 qsort(res_arr,res_arr_cnt,sizeof(int),compare); //释放内存 for(i = 0;i < num;i++){ printf("%d ",res_arr[i]); free(arr[i]); } free(arr); }