Hướng dẫn cho Đếm cặp

Chỉ sử dụng khi thực sự cần thiết như một cách tôn trọng tác giả và người viết hướng dẫn này.

Chép code từ bài hướng dẫn để nộp bài là hành vi có thể dẫn đến khóa tài khoản.

Authors: SPyofgame

\(\color{red}{\text{Spoiler Alert}_{{}_{{}^{{}^{v2.0}}}}}\)

\(\color{red}{\text{Khuyến khích bạn đọc trước khi đọc phần lời giải xin hãy thử code ra thuật của mình dù nó có sai hay đúng}}\)

\(\color{red}{\text{Sau đó từ phần bài giải và thuật toán trước đó mà đối chiếu, rút nhận xét với thuật của mình và thu được bài học (không lãng phí thời gian đâu).}}\)

\(\color{orange}{\text{Hint 1 <Brute-force>}}\)

  • Duyệt từng cặp \((i, j)\) và đếm xem có bao nhiêu cặp thỏa mãn

\(\color{orange}{\text{Hint 2 <STL>::<Map> <Sorting> <Online Solving>}}\)

  • Có \(a_i + a_j^2 = k \Leftrightarrow a_i = k - a_j^2\)

Tại mỗi vị trí \(j\), ta nhận phần tử và tăng số lần xuất hiện của nó

Sau đó ta tìm \(a_i = k - a_j^2\) và đếm xem \(a_i\) xuất hiện bao nhiêu lần

\(\color{green}{\text{Preference AC Code }}\): STL::map, Sorting, Online Solving

\(^{^{\color{purple}{\text{Complexity : }} O(n \log n)\ \color{purple}{\text{time}}\ ||\ O(n)\ \color{purple}{\text{memory}}}}\)

C++
int main()
{
    int n, k;
    cin >> n >> k;

    ll res = 0;
    map<int, int> M;
    for (int i = 0; i < n; ++i)
    {
        int x;
        cin >> x;
        M[x]++;
        res += M[k - 1LL * x * x];
    }
    cout << res;
    return 0;
}

Bình luận