#define NULL ((void *)0)
typedef unsigned long size_t; // Customize by platform.
typedef long intptr_t;
typedef unsigned long uintptr_t;
typedef long scalar_t__; // Either arithmetic or pointer type.
/* By default, we understand bool (as a convenience). */
typedef int bool;
#define false 0
#define true 1

/* Forward declarations */

/* Type definitions */

/* Variables and functions */
scalar_t__ malloc(int);
int *distributeCandies(int candies, int num_people, int *returnSize) {
  int *arr = (int *)malloc(sizeof(int) * num_people);
  int n = num_people, k = 0;
  int left = candies;
  int cd = 1;
  *returnSize = num_people;
  for (int i = 0; i < n; i++)
    arr[i] = 0;
  int i = 0;
  while (candies > 0) {
    if (candies >= cd)
      arr[i++] += cd++;
    else {
      arr[i] += candies;
      break;
    }
    candies -= (cd - 1);
    if (i == num_people)
      i = 0;
  }
  return arr;
}


x86 assembly (clang, -O0):
<distributeCandies>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %edi,-0x4(%rbp)
mov    %esi,-0x8(%rbp)
mov    %rdx,-0x10(%rbp)
movslq -0x8(%rbp),%rax
shl    $0x2,%rax
mov    %eax,%edi
call   1030 <malloc@plt>
mov    %rax,-0x18(%rbp)
mov    -0x8(%rbp),%eax
mov    %eax,-0x1c(%rbp)
movl   $0x0,-0x20(%rbp)
mov    -0x4(%rbp),%eax
mov    %eax,-0x24(%rbp)
movl   $0x1,-0x28(%rbp)
mov    -0x8(%rbp),%ecx
mov    -0x10(%rbp),%rax
mov    %ecx,(%rax)
movl   $0x0,-0x2c(%rbp)
mov    -0x2c(%rbp),%eax
cmp    -0x1c(%rbp),%eax
jge    1188 <distributeCandies+0x78>
mov    -0x18(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movl   $0x0,(%rax,%rcx,4)
mov    -0x2c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x2c(%rbp)
jmp    115f <distributeCandies+0x4f>
movl   $0x0,-0x30(%rbp)
cmpl   $0x0,-0x4(%rbp)
jle    1209 <distributeCandies+0xf9>
mov    -0x4(%rbp),%eax
cmp    -0x28(%rbp),%eax
jl     11cd <distributeCandies+0xbd>
mov    -0x28(%rbp),%edx
mov    %edx,%eax
add    $0x1,%eax
mov    %eax,-0x28(%rbp)
mov    -0x18(%rbp),%rax
mov    -0x30(%rbp),%ecx
mov    %ecx,%esi
add    $0x1,%esi
mov    %esi,-0x30(%rbp)
movslq %ecx,%rcx
add    (%rax,%rcx,4),%edx
mov    %edx,(%rax,%rcx,4)
jmp    11e3 <distributeCandies+0xd3>
mov    -0x4(%rbp),%edx
mov    -0x18(%rbp),%rax
movslq -0x30(%rbp),%rcx
add    (%rax,%rcx,4),%edx
mov    %edx,(%rax,%rcx,4)
jmp    1209 <distributeCandies+0xf9>
mov    -0x28(%rbp),%ecx
sub    $0x1,%ecx
mov    -0x4(%rbp),%eax
sub    %ecx,%eax
mov    %eax,-0x4(%rbp)
mov    -0x30(%rbp),%eax
cmp    -0x8(%rbp),%eax
jne    1204 <distributeCandies+0xf4>
movl   $0x0,-0x30(%rbp)
jmp    118f <distributeCandies+0x7f>
mov    -0x18(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<distributeCandies>:
push   %r15
push   %r14
push   %r12
push   %rbx
push   %rax
mov    %rdx,%r12
mov    %esi,%r14d
mov    %edi,%ebx
lea    0x0(,%r14,4),%edi
call   1040 <malloc@plt>
mov    %rax,%r15
mov    %r14d,(%r12)
test   %r14d,%r14d
jle    115a <distributeCandies+0x3a>
mov    %r14d,%edx
shl    $0x2,%rdx
mov    %r15,%rdi
xor    %esi,%esi
call   1030 <memset@plt>
test   %ebx,%ebx
jle    1193 <distributeCandies+0x73>
mov    $0x1,%ecx
xor    %eax,%eax
xor    %edx,%edx
nopw   0x0(%rax,%rax,1)
movslq %edx,%rdx
cmp    %ecx,%ebx
jl     118f <distributeCandies+0x6f>
add    %ecx,(%r15,%rdx,4)
lea    0x1(%rcx),%esi
add    $0x1,%edx
cmp    %r14d,%edx
cmove  %eax,%edx
sub    %ecx,%ebx
mov    %esi,%ecx
jg     1170 <distributeCandies+0x50>
jmp    1193 <distributeCandies+0x73>
add    %ebx,(%r15,%rdx,4)
mov    %r15,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
ret

LLVM IR (-O0):
define i32* @distributeCandies(i32 %0, i32 %1, i32* %2) {
%4 = alloca i32
%5 = alloca i32
%6 = alloca i32*
%7 = alloca i32*
%8 = alloca i32
%9 = alloca i32
%10 = alloca i32
%11 = alloca i32
%12 = alloca i32
%13 = alloca i32
store i32 %0, i32* %4
store i32 %1, i32* %5
store i32* %2, i32** %6
%14 = load i32, i32* %5
%15 = sext i32 %14 to i64
%16 = mul i64 4, %15
%17 = trunc i64 %16 to i32
%18 = call i64 @malloc(i32 %17)
%19 = inttoptr i64 %18 to i32*
store i32* %19, i32** %7
%20 = load i32, i32* %5
store i32 %20, i32* %8
store i32 0, i32* %9
%21 = load i32, i32* %4
store i32 %21, i32* %10
store i32 1, i32* %11
%22 = load i32, i32* %5
%23 = load i32*, i32** %6
store i32 %22, i32* %23
store i32 0, i32* %12
br label %24
24: ; preds = %33, %3
%25 = load i32, i32* %12
%26 = load i32, i32* %8
%27 = icmp slt i32 %25, %26
br i1 %27, label %28, label %36
28: ; preds = %24
%29 = load i32*, i32** %7
%30 = load i32, i32* %12
%31 = sext i32 %30 to i64
%32 = getelementptr inbounds i32, i32* %29, i64 %31
store i32 0, i32* %32
br label %33
33: ; preds = %28
%34 = load i32, i32* %12
%35 = add nsw i32 %34, 1
store i32 %35, i32* %12
br label %24
36: ; preds = %24
store i32 0, i32* %13
br label %37
37: ; preds = %71, %36
%38 = load i32, i32* %4
%39 = icmp sgt i32 %38, 0
br i1 %39, label %40, label %72
40: ; preds = %37
%41 = load i32, i32* %4
%42 = load i32, i32* %11
%43 = icmp sge i32 %41, %42
br i1 %43, label %44, label %54
44: ; preds = %40
%45 = load i32, i32* %11
%46 = add nsw i32 %45, 1
store i32 %46, i32* %11
%47 = load i32*, i32** %7
%48 = load i32, i32* %13
%49 = add nsw i32 %48, 1
store i32 %49, i32* %13
%50 = sext i32 %48 to i64
%51 = getelementptr inbounds i32, i32* %47, i64 %50
%52 = load i32, i32* %51
%53 = add nsw i32 %52, %45
store i32 %53, i32* %51
br label %62
54: ; preds = %40
%55 = load i32, i32* %4
%56 = load i32*, i32** %7
%57 = load i32, i32* %13
%58 = sext i32 %57 to i64
%59 = getelementptr inbounds i32, i32* %56, i64 %58
%60 = load i32, i32* %59
%61 = add nsw i32 %60, %55
store i32 %61, i32* %59
br label %72
62: ; preds = %44
%63 = load i32, i32* %11
%64 = sub nsw i32 %63, 1
%65 = load i32, i32* %4
%66 = sub nsw i32 %65, %64
store i32 %66, i32* %4
%67 = load i32, i32* %13
%68 = load i32, i32* %5
%69 = icmp eq i32 %67, %68
br i1 %69, label %70, label %71
70: ; preds = %62
store i32 0, i32* %13
br label %71
71: ; preds = %70, %62
br label %37
72: ; preds = %54, %37
%73 = load i32*, i32** %7
ret i32* %73
}

LLVM IR (-O2):
define i32* @distributeCandies(i32 %0, i32 %1, i32* nocapture writeonly %2) {
%4 = shl i32 %1, 2
%5 = tail call i64 @malloc(i32 %4)
%6 = inttoptr i64 %5 to i32*
store i32 %1, i32* %2
%7 = icmp sgt i32 %1, 0
br i1 %7, label %8, label %12
8: ; preds = %3
%9 = inttoptr i64 %5 to i8*
%10 = zext i32 %1 to i64
%11 = shl nuw nsw i64 %10, 2
br label %12
12: ; preds = %8, %3
%13 = icmp sgt i32 %0, 0
br i1 %13, label %14, label %35
14: ; preds = %12, %19
%15 = phi i32 [ %28, %19 ], [ 0, %12 ]
%16 = phi i32 [ %20, %19 ], [ 1, %12 ]
%17 = phi i32 [ %26, %19 ], [ %0, %12 ]
%18 = icmp slt i32 %17, %16
br i1 %18, label %30, label %19
19: ; preds = %14
%20 = add nuw nsw i32 %16, 1
%21 = add nsw i32 %15, 1
%22 = sext i32 %15 to i64
%23 = getelementptr inbounds i32, i32* %6, i64 %22
%24 = load i32, i32* %23
%25 = add nsw i32 %24, %16
store i32 %25, i32* %23
%26 = sub nsw i32 %17, %16
%27 = icmp eq i32 %21, %1
%28 = select i1 %27, i32 0, i32 %21
%29 = icmp sgt i32 %26, 0
br i1 %29, label %14, label %35
30: ; preds = %14
%31 = sext i32 %15 to i64
%32 = getelementptr inbounds i32, i32* %6, i64 %31
%33 = load i32, i32* %32
%34 = add nsw i32 %33, %17
store i32 %34, i32* %32
br label %35
35: ; preds = %19, %12, %30
ret i32* %6
}