giúp mình với mình đang cần gấp

#include <bits/stdc++.h>

using namespace std;
#define int long long

#define endl "\n"

signed main(){

    freopen("BAI2.INP", "r", stdin);

    freopen("BAI2.OUT", "w", stdout);

    int n; cin >> n; // nhap vao so phan tu cua mang

    int a[n + 3]; // khai bao mang kieu so nguyen 32-bit

    for (int i = 1; i <= n; i++){

        cin >> a[i]; // nhap vao phan tu cua mang

    }

    sort(a + 1, a + n + 1); // sap xep bang ham co san

    for (int i = 1; i <= n; i++){

        cout << a[i] << " "; // in ra phan tu

    }

    return 0; // ket thuc chuong trinh

    // do phuc tap thoi gian: O(nlogn)

}

def gt_tong(n):
    return tong(map(int, str(n)))

with open('BAI2.INP', 'r') as inp:
    n = int(inp.readline().strip())
    so = list(map(int, inp.readline().strip().split()))

so_2 = sorted(nums, key=lambda x: (digit_sum(x), -x))

with open('BAI2.OUT', 'w') as out:
    out.write(' '.join(map(str, sorted_nums)))