C++程序测试用例验收

Question

出于好奇，我试图解决编码竞赛。 在那里，有一个问题无法通过我的解决方案的所有测试用例。 您认为我的解决方案是否有任何改进可能会有所帮助？ TIA。 下面说明了问题以及我的解决方案。 约束：

1 ≤ M,N ≤ 10^7,
1 ≤ a[i],b[i] ≤ 10^7

Output：

打印一个int，他用给定的金额购买两件物品后可以获得的最大酷值

我的解决方案是：

#include<bits/stdc++.h>
using namespace std;

int main()
{

    long long N, M, C, count;

    cin>>N>>M>>C;

    long long a[N], b[M];

    for(long long i=0; i<N; i++){
        cin>>a[i];
    }

    for(long long i=0; i<M; i++){
        cin>>b[i];
    }

    sort(a, a+N);
    sort(b, b+M);


    if(a[N-1] + b[M-1] <= C){
        count = (N-1) + (M-1);
        cout<<count<<endl;
    }
    else{
        cout<<"Not even closer"<<endl;
    }

    return 0;
}

Can anyone suggest what I could improve here to pass all the test? 

Answer 1

排序后，价格a[i]的酷度等于i 。

因此，解决方案包括对a和b进行排序，然后最大化i+j使得a[i]+a[j] <= C

最后的优化是由一个简单for循环执行的，有两个索引，一个用于a ，一个用于b

复杂度以排序为主：O(NlogN + MlogM)

注意：问题中当前贴出的代码是经过一些评论后编辑的，与原代码不对应

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(NULL);

    int N, M, C;
    std::cin >> N >> M >> C;
    std::vector<int> a(N), b(M);

    for (int i = 0; i < N; i++){
        std::cin >> a[i];
    }

    for (int i = 0; i < M; i++){
        std::cin >> b[i];
    }

    sort(a.begin(), a.end());
    sort(b.begin(), b.end());

    if(a[0] + b[0] > C) {
        std::cout <<"E kemon aynabaji!!!" << std::endl;
        return 0;
    }
    
    int max_cool = 0;
    int iB = M-1;
    for (int iA = 0; iA < N; ++iA) {
        while (iB >= 0 && (a[iA] + b[iB] > C)) {
            iB--;
        }
        if (iB < 0) break;
        int cool = iA + iB;
        if (cool > max_cool) max_cool = cool;
    }

    std::cout << max_cool << std::endl;
    return 0;
}