aboutsummaryrefslogtreecommitdiff
path: root/replace_non_coprime_nums.c
blob: 2981ceea13a8b56d981bd927a5c31cd6d65e9f98 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// 2197. Replace Non-Coprime Numbers in Array
#include "leetcode.h"

/*
 * you are given an array of integers 'nums'. perform the following steps: find
 * any two adjacent numbers in 'nums' that are non-coprime, if no such numbers
 * are found, stop the process. otherwise, delete the two numbers and replace
 * them with their lcm. repeat this process as long as you keep finding two
 * adjacent non-coprime integers. return the final modified array. it can be
 * shown that replacing adjacent non-coprime numbers in any arbitrary order will
 * lead to the same result. two values 'x' and '' aree non-coprime if 'gcd(x, y)
 * > 1' where 'gcd(x, y)' is the greatest common denomiator
 */

int gcd(int a, int b) {
  while (b != 0) {
    int tmp = b;
    b = a % b;
    a = tmp;
  }
  return a;
}

int *replaceNonCoprimes(int *nums, int numsSize, int *returnSize) {
  int *ans = (int *)malloc(numsSize * sizeof(int)), top = 0;
  for (int i = 0; i < numsSize; i++) {
    int curr = nums[i];
    while (top > 0) {
      int prev = ans[top - 1];
      int g = gcd(prev, curr);
      if (g == 1)
        break;
      top--;
      long long merged = (long long)prev / g * curr;
      curr = (int)merged;
    }
    ans[top++] = curr;
  }
  *returnSize = top;
  return ans;
}

int main() {
  int n1[] = {6, 4, 3, 2, 7, 6, 2}, rs1;
  int n2[] = {2, 2, 1, 1, 3, 3, 3}, rs2;
  int *rnc1 = replaceNonCoprimes(n1, ARRAY_SIZE(n1), &rs1);
  int *rnc2 = replaceNonCoprimes(n2, ARRAY_SIZE(n2), &rs2);
  for (int i = 0; i < rs1; i++)
    printf("%d ", rnc1[i]); // expect: 12 7 6
  printf("\n");
  for (int i = 0; i < rs2; i++)
    printf("%d ", rnc2[i]); // expect: 2 1 1 3
  printf("\n");
  free(rnc1);
  free(rnc2);
}