python 多参数排序

好的，这是一个关于股票的程序，其中有两个类: 深圳类 shen_stock 和上海类 shang_stock。它们都有三个私有数据成员：普通股票个数 general、ST股票个数 st 和 PT股票个数 pt。两个类分别有自己的友元函数来计算并显示深圳或上海的股票总数（三项的和）。两个类还共用一个 count() 函数，用来计算深圳和上海总共有多少股票并输出。程序需要使用多文件编程实现。

以下是程序示例：

shen_stock.h:

#ifndef SHEN_STOCK_H
#define SHEN_STOCK_H

class shen_stock {
private:
    int general;
    int st;
    int pt;
public:
    shen_stock();
    void add_stock(int type, int amount);
friend int shen_total(shen_stock &s);
friend int count_total(shen_stock &s, shang_stock &h);
};

#endif

shen_stock.cpp:

#include <iostream>
using namespace std;
#include "shen_stock.h"

shen_stock::shen_stock() {
    general = 0;
    st = 0;
    pt = 0;
}

void shen_stock::add_stock(int type, int amount) {
    switch(type) {
        case 0:
            general += amount;
            break;
        case 1:
            st += amount;
            break;
        case 2:
            pt += amount;
            break;
        default:
            cout << "Invalid stock type!" << endl;
    }
}

int shen_total(shen_stock &s