Question

我正在编写一个简单的程序来计算数组中可被3个数组长度整除的数量，并且值由用户确定。

现在我的代码非常好。但是，我只想检查是否有更快的方法来计算它，从而减少编译时间？

由于数组的长度为10 ^ 4或更少，编译器所需的时间不到100毫秒。然而，随着它增加到10 ^ 5，它会达到1000毫秒，所以为什么呢？以及如何提高速度？

#include <iostream>

using namespace std;

int main()
{
    int N, i, b;
    b = 0;
    cin >> N;

    unsigned int j = 0;
    std::vector<unsigned int> a(N);
    for (j = 0; j < N; j++) {
        cin >> a[j];
        if (j == 0) {
        }

        else {
            for (i = j - 1; i >= 0; i = i - 1) {
                if ((a[j] + a[i]) % 3 == 0) {
                    b++;
                }
            }
        }
    }

    cout << b;

    return 0;
}

Answer 1

您的算法具有O(N^2)复杂度。有一种更快的方法。

(a[i] + a[j]) % 3 == ((a[i] % 3) + (a[j] % 3)) % 3

因此，你不需要知道确切的数字，你需要知道它们的除法余数只有三个。余数的零余数可以使用两个余数为(0 + 0)的数字以及两个余数为1和2 (1 + 2)的数字来接收。

结果将等于r[1]*r[2] + r[0]*(r[0]-1)/2，其中r[i]是余数等于i的数字数量。

int r[3] = {};
for (int i : a) {
    r[i % 3]++; 
}
std::cout << r[1]*r[2] + (r[0]*(r[0]-1)) / 2;

此算法的复杂性为O(N)。

Answer 2

之前我遇到过这个问题，虽然我找不到特定的解决方案，但您可以通过散列来改善运行时间。

代码看起来像这样：

// A C++ program to check if arr[0..n-1] can be divided
// in pairs such that every pair is divisible by k.
#include <bits/stdc++.h>
using namespace std;

// Returns true if arr[0..n-1] can be divided into pairs
// with sum divisible by k.
bool canPairs(int arr[], int n, int k)
{
    // An odd length array cannot be divided into pairs
    if (n & 1)
         return false;

    // Create a frequency array to count occurrences
    // of all remainders when divided by k.
    map<int, int> freq;

    // Count occurrences of all remainders
    for (int i = 0; i < n; i++)
        freq[arr[i] % k]++;

    // Traverse input array and use freq[] to decide
    // if given array can be divided in pairs
    for (int i = 0; i < n; i++)
    {
        // Remainder of current element
        int rem = arr[i] % k;

        // If remainder with current element divides
        // k into two halves.
        if  (2*rem == k)
        {
            // Then there must be even occurrences of
            // such remainder
            if (freq[rem] % 2 != 0)
                return false;
        }

        // If remainder is 0, then there must be two 
        // elements with 0 remainder
        else if (rem == 0)
        {
           if (freq[rem] & 1)           
               return false;
        }        

        // Else number of occurrences of remainder
        // must be equal to number of occurrences of
        // k - remainder
        else if (freq[rem] != freq[k - rem])
            return false;
    }
    return true;
}

/* Driver program to test above function */
int main()
{
    int arr[] =  {92, 75, 65, 48, 45, 35};
    int k = 10;
    int n = sizeof(arr)/sizeof(arr[0]);
    canPairs(arr, n, k)? cout << "True": cout << "False";
    return 0;
}

适用于k（在您的情况下为3）但话说回来，这不是我的代码，而是您可以在following link.中找到的代码并给出正确的解释。我不只是粘贴链接，因为我认为这是不好的做法。

改进嵌套循环的优化

2 个答案: