用于计算条件为真的实例的代码

Question

最近我在为即将到来的zoc练习时遇到了this codechef问题。

问题要求计算给定数字的组合数，其中总和大于另一个给定数字。

以前我有一个暴力算法给出了正确的答案，但是花了很多时间，在某些情况下甚至超过了一秒钟，它在codechef中引发了TLE错误。

所以我改变了我的整个算法以摆脱那个TLE错误，但是现在它只给出了第一次测试的正确答案，其余的产生了错误的输出，有人可以帮我弄清楚我在做错了什么：< / p>

#include <iostream>
#include <vector>
int main(){
    std::vector<long long> tstCases;
    std::vector<long long> okset;
    std::vector<long long> testset;
    for(int i =0;i<2;i++){
        long long cases;
        std::cin >> cases;
        tstCases.push_back(cases);
    }
    long long n = tstCases.at(0);
    long long k = tstCases.at(1);
    for(long long i =0;i<n;i++){
        long long cases;
        std::cin >> cases;
        if(cases<=k){
            if(cases < k/2){
                okset.push_back(cases);
            }else{
                testset.push_back(cases);
            }
        }
    }
    long long l = okset.size();
    long long p1 = (l*(l-1))/2;
    long long p2 = 0;
    if(l > 0){
        for(long long i =0;i<l;i++){
            for(long long j =0;j<testset.size();j++){
                if(okset.at(i)+testset.at(j) >= k){
                    break;
                }else{
                    p2++;
                }
            }
        }
    }
    long long p = p1+p2;
    std::cout << p << std::endl;
    return 0;

}

注意：我正在使用long long，因为问题指示我这样做，而且我很懒，我自己这样做，所以我使用了替换全部。对不起。

Answer 1

好吧，看起来我没有对testset数组进行排序，因此很多项都留在了break语句之后。感谢uSeemSurprised帮助我注意到这一点。