欠落している数がx
で、重複が であるとしy
ます。すべての数値を合計すると、次のようになります。
(n - 1) * n / 2 - x + y
以上から、(x - y)
・・・・(1)
同様に、数値の 2 乗を合計します。合計は次のようになります。
(n - 1) * n * (2 * n - 1) / 6 - x2 + y2
上記から、 ....(2)が得られます。(x2 - y2)
(2) / (1) gives (x + y).....(3)
(1) + (3) が与えられ、それによってとが得2 * x
られます。x
y
このソリューションにはO(1)
余分なストレージがあり、O(n)
時間の複雑さに注意してください。上記の他の解決策は、不必要にO(n)
余分なストレージです。
より明確にするために、混合 C/C++ でコードを記述します。
#include <stdio.h>
int findDup(int *arr, int n, int& dup, int& missing)
{
int sum = 0;
int squares = 0;
for (int i = 0; i < n; i++) {
sum += arr[i];
squares += arr[i] * arr[i];
}
sum = (n - 1) * n / 2 - sum; // x - y
squares = (n - 1) * n * (2 * (n - 1) + 1) / 6 - squares; // x^2 - y^2
if (sum == 0) {
// no duplicates
missing = dup = 0;
return -1;
}
missing = (squares / sum + sum) / 2; // ((x^2 - y^2) / (x - y) + (x - y)) / 2 = ((x + y) + (x - y)) / 2 = x
dup = missing - sum; // x - (x - y) = y
return 0;
}
int main(int argc, char *argv[])
{
int dup = 0;
int missing = 0;
int a[] = {0, 2, 1, 2, 4};
findDup(a, sizeof(a) / sizeof(int), dup, missing);
printf("dup = [%d], missing = [%d]\n", dup, missing);
int b[] = {3, 0, 0, 4, 2, 1};
findDup(b, sizeof(b) / sizeof(int), dup, missing);
printf("dup = [%d], missing = [%d]\n", dup, missing);
return 0;
}
出力:
dup = [2], missing = [3]
dup = [0], missing = [5]
いくつかの python コード:
def finddup(lst):
sum = 0
sumsq = 0
missing = 0
dup = 0
for item in lst:
sum = sum + item
sumsq = sumsq + item * item
n = len(a)
sum = (n - 1) * n / 2 - sum
sumsq = (n - 1) * n * (2 * (n - 1) + 1) / 6 - sumsq
if sum == 0:
return [-1, missing, dup]
missing = ((sumsq / sum) + sum) / 2
dup = missing - sum
return [0, missing, dup]
found, missing, dup = finddup([0, 2, 1, 2, 4])
if found != -1:
print "dup = " + str(dup) + " missing = " + str(missing)
print finddup([3, 0, 0, 4, 2, 1])
出力:
dup = 2 missing = 3
[-1, 0, 0]