aboutsummaryrefslogtreecommitdiff
path: root/convert_arr_2d_cond.c
blob: 9e3effc8862c8c715116b18d971a6da5cfce4dac (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
// 2610. Convert an Array Into a 2D Array With Conditions
#include "leetcode.h"

/*
 * given an integer array 'nums', you should create a 2d array from 'nums'
 * satisfying the following conditions. the 2d array should contain only the
 * elements of the array 'nums'. each row in the 2d array contains distinct
 * integers. the number of rows in the 2d array should be minimal. return the
 * resulting array. if there are multiple answers, return any of them. note that
 * the 2d array can have a different number of elements on each row.
 */

int **findMatrix(int *nums, int nums_size, int *return_size,
                 int **return_column_sizes) {
  int *table = calloc(201, sizeof(int));
  *return_size = 0;
  for (int i = 0; i < nums_size; i++) {
    table[nums[i]]++;
    *return_size = fmax(*return_size, table[nums[i]]);
  }
  int **ans = malloc(sizeof(int *) * *return_size);
  *return_column_sizes = calloc(*return_size, sizeof(int));
  for (int i = 0; i < *return_size; i++)
    ans[i] = calloc(nums_size, sizeof(int));
  for (int i = 0; i <= 200; i++) {
    int row = 0;
    while (table[i]) {
      ans[row][*return_column_sizes[row]++] = i;
      table--;
      row++;
    }
  }
  free(table);
  return ans;
}