độ phức tạp `O(n)`

giải = python bị TLE với `k >= 10^7`

import math as m
def main(k):
    ans = 0
    for canh1 in range(1,k,1):
        canh2 = m.sqrt(k**2-canh1**2)
        if int(canh2) == canh2:
           ans = max(canh1*canh2,ans)
    return ans
k = int(input())
kq = int(main(k))
print(kq)

giải bằng cpp cũng có độ phức tạp `O(n)` nhưng AC

#include <iostream>
#include <sstream>
#include <cmath>
#include <iomanip>
using namespace std;
using ll = long long;
int main(){
    int k; cin >> k;
    double kq, canh2;
    for (int canh1 = 1;canh1 < k;canh1 += 1){
        canh2 = sqrt(pow(k,2)-pow(canh1,2));
        if (ll(canh2) == canh2){
            kq = max(canh1*canh2,kq);
        }
    }
    stringstream ss;
    ss << fixed << setprecision(0) << kq;
    ll result;
    ss >> result;
    cout << result << endl;
}