#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 *InitUnionFind(int n) {
  int *set = (int *)malloc(n * sizeof(int));
  for (int i = 0; i < n; i++) {
    set[i] = i;
  }
  return set;
}


x86 assembly (clang, -O0):
<InitUnionFind>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %edi,-0x4(%rbp)
movslq -0x4(%rbp),%rax
shl    $0x2,%rax
mov    %eax,%edi
call   1030 <malloc@plt>
mov    %rax,-0x10(%rbp)
movl   $0x0,-0x14(%rbp)
mov    -0x14(%rbp),%eax
cmp    -0x4(%rbp),%eax
jge    115d <InitUnionFind+0x4d>
mov    -0x14(%rbp),%edx
mov    -0x10(%rbp),%rax
movslq -0x14(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x14(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x14(%rbp)
jmp    1135 <InitUnionFind+0x25>
mov    -0x10(%rbp),%rax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<InitUnionFind>:
push   %rbx
mov    %edi,%ebx
lea    0x0(,%rbx,4),%edi
call   1030 <malloc@plt>
test   %ebx,%ebx
jle    129c <InitUnionFind+0x18c>
mov    %ebx,%ecx
cmp    $0x8,%ebx
jae    1135 <InitUnionFind+0x25>
xor    %edx,%edx
jmp    1290 <InitUnionFind+0x180>
mov    %ecx,%edx
and    $0xfffffff8,%edx
lea    -0x8(%rdx),%rbx
mov    %rbx,%rdi
shr    $0x3,%rdi
add    $0x1,%rdi
mov    %edi,%esi
and    $0x3,%esi
cmp    $0x18,%rbx
jae    1163 <InitUnionFind+0x53>
movdqa 0xea4(%rip),%xmm0
xor    %ebx,%ebx
jmp    1238 <InitUnionFind+0x128>
and    $0xfffffffffffffffc,%rdi
movdqa 0xe91(%rip),%xmm0
xor    %ebx,%ebx
movdqa 0xe96(%rip),%xmm8
movdqa 0xe9d(%rip),%xmm9
movdqa 0xea4(%rip),%xmm10
movdqa 0xeac(%rip),%xmm4
movdqa 0xeb4(%rip),%xmm5
movdqa 0xebc(%rip),%xmm6
movdqa 0xec4(%rip),%xmm7
movdqa 0xecc(%rip),%xmm1
cs nopw 0x0(%rax,%rax,1)
xchg   %ax,%ax
movdqa %xmm0,%xmm2
paddd  %xmm8,%xmm2
movdqu %xmm0,(%rax,%rbx,4)
movdqu %xmm2,0x10(%rax,%rbx,4)
movdqa %xmm0,%xmm2
paddd  %xmm9,%xmm2
movdqa %xmm0,%xmm3
paddd  %xmm10,%xmm3
movdqu %xmm2,0x20(%rax,%rbx,4)
movdqu %xmm3,0x30(%rax,%rbx,4)
movdqa %xmm0,%xmm2
paddd  %xmm4,%xmm2
movdqa %xmm0,%xmm3
paddd  %xmm5,%xmm3
movdqu %xmm2,0x40(%rax,%rbx,4)
movdqu %xmm3,0x50(%rax,%rbx,4)
movdqa %xmm0,%xmm2
paddd  %xmm6,%xmm2
movdqa %xmm0,%xmm3
paddd  %xmm7,%xmm3
movdqu %xmm2,0x60(%rax,%rbx,4)
movdqu %xmm3,0x70(%rax,%rbx,4)
add    $0x20,%rbx
paddd  %xmm1,%xmm0
add    $0xfffffffffffffffc,%rdi
jne    11c0 <InitUnionFind+0xb0>
test   %rsi,%rsi
je     1280 <InitUnionFind+0x170>
lea    (%rax,%rbx,4),%rdi
add    $0x10,%rdi
shl    $0x5,%rsi
xor    %ebx,%ebx
movdqa 0xdbd(%rip),%xmm1
movdqa 0xdc5(%rip),%xmm2
nopl   0x0(%rax,%rax,1)
movdqa %xmm0,%xmm3
paddd  %xmm1,%xmm3
movdqu %xmm0,-0x10(%rdi,%rbx,1)
movdqu %xmm3,(%rdi,%rbx,1)
paddd  %xmm2,%xmm0
add    $0x20,%rbx
cmp    %rbx,%rsi
jne    1260 <InitUnionFind+0x150>
cmp    %rcx,%rdx
je     129c <InitUnionFind+0x18c>
cs nopw 0x0(%rax,%rax,1)
nop
mov    %edx,(%rax,%rdx,4)
add    $0x1,%rdx
cmp    %rdx,%rcx
jne    1290 <InitUnionFind+0x180>
pop    %rbx
ret

LLVM IR (-O0):
define i32* @InitUnionFind(i32 %0) {
%2 = alloca i32
%3 = alloca i32*
%4 = alloca i32
store i32 %0, i32* %2
%5 = load i32, i32* %2
%6 = sext i32 %5 to i64
%7 = mul i64 %6, 4
%8 = trunc i64 %7 to i32
%9 = call i64 @malloc(i32 %8)
%10 = inttoptr i64 %9 to i32*
store i32* %10, i32** %3
store i32 0, i32* %4
br label %11
11: ; preds = %21, %1
%12 = load i32, i32* %4
%13 = load i32, i32* %2
%14 = icmp slt i32 %12, %13
br i1 %14, label %15, label %24
15: ; preds = %11
%16 = load i32, i32* %4
%17 = load i32*, i32** %3
%18 = load i32, i32* %4
%19 = sext i32 %18 to i64
%20 = getelementptr inbounds i32, i32* %17, i64 %19
store i32 %16, i32* %20
br label %21
21: ; preds = %15
%22 = load i32, i32* %4
%23 = add nsw i32 %22, 1
store i32 %23, i32* %4
br label %11
24: ; preds = %11
%25 = load i32*, i32** %3
ret i32* %25
}

LLVM IR (-O2):
define i32* @InitUnionFind(i32 %0) {
%2 = shl i32 %0, 2
%3 = tail call i64 @malloc(i32 %2)
%4 = inttoptr i64 %3 to i32*
%5 = icmp sgt i32 %0, 0
br i1 %5, label %6, label %73
6: ; preds = %1
%7 = zext i32 %0 to i64
%8 = icmp ult i32 %0, 8
br i1 %8, label %71, label %9
9: ; preds = %6
%10 = and i64 %7, 4294967288
%11 = add nsw i64 %10, -8
%12 = lshr exact i64 %11, 3
%13 = add nuw nsw i64 %12, 1
%14 = and i64 %13, 3
%15 = icmp ult i64 %11, 24
br i1 %15, label %52, label %16
16: ; preds = %9
%17 = and i64 %13, 4611686018427387900
br label %18
18: ; preds = %18, %16
%19 = phi i64 [ 0, %16 ], [ %48, %18 ]
%20 = phi <4 x i32> [ <i32 0, i32 1, i32 2, i32 3>, %16 ], [ %49, %18 ]
%21 = phi i64 [ 0, %16 ], [ %50, %18 ]
%22 = add <4 x i32> %20, <i32 4, i32 4, i32 4, i32 4>
%23 = getelementptr inbounds i32, i32* %4, i64 %19
%24 = bitcast i32* %23 to <4 x i32>*
store <4 x i32> %20, <4 x i32>* %24
%25 = getelementptr inbounds i32, i32* %23, i64 4
%26 = bitcast i32* %25 to <4 x i32>*
store <4 x i32> %22, <4 x i32>* %26
%27 = or i64 %19, 8
%28 = add <4 x i32> %20, <i32 8, i32 8, i32 8, i32 8>
%29 = add <4 x i32> %20, <i32 12, i32 12, i32 12, i32 12>
%30 = getelementptr inbounds i32, i32* %4, i64 %27
%31 = bitcast i32* %30 to <4 x i32>*
store <4 x i32> %28, <4 x i32>* %31
%32 = getelementptr inbounds i32, i32* %30, i64 4
%33 = bitcast i32* %32 to <4 x i32>*
store <4 x i32> %29, <4 x i32>* %33
%34 = or i64 %19, 16
%35 = add <4 x i32> %20, <i32 16, i32 16, i32 16, i32 16>
%36 = add <4 x i32> %20, <i32 20, i32 20, i32 20, i32 20>
%37 = getelementptr inbounds i32, i32* %4, i64 %34
%38 = bitcast i32* %37 to <4 x i32>*
store <4 x i32> %35, <4 x i32>* %38
%39 = getelementptr inbounds i32, i32* %37, i64 4
%40 = bitcast i32* %39 to <4 x i32>*
store <4 x i32> %36, <4 x i32>* %40
%41 = or i64 %19, 24
%42 = add <4 x i32> %20, <i32 24, i32 24, i32 24, i32 24>
%43 = add <4 x i32> %20, <i32 28, i32 28, i32 28, i32 28>
%44 = getelementptr inbounds i32, i32* %4, i64 %41
%45 = bitcast i32* %44 to <4 x i32>*
store <4 x i32> %42, <4 x i32>* %45
%46 = getelementptr inbounds i32, i32* %44, i64 4
%47 = bitcast i32* %46 to <4 x i32>*
store <4 x i32> %43, <4 x i32>* %47
%48 = add nuw i64 %19, 32
%49 = add <4 x i32> %20, <i32 32, i32 32, i32 32, i32 32>
%50 = add i64 %21, 4
%51 = icmp eq i64 %50, %17
br i1 %51, label %52, label %18
52: ; preds = %18, %9
%53 = phi i64 [ 0, %9 ], [ %48, %18 ]
%54 = phi <4 x i32> [ <i32 0, i32 1, i32 2, i32 3>, %9 ], [ %49, %18 ]
%55 = icmp eq i64 %14, 0
br i1 %55, label %69, label %56
56: ; preds = %52, %56
%57 = phi i64 [ %65, %56 ], [ %53, %52 ]
%58 = phi <4 x i32> [ %66, %56 ], [ %54, %52 ]
%59 = phi i64 [ %67, %56 ], [ 0, %52 ]
%60 = add <4 x i32> %58, <i32 4, i32 4, i32 4, i32 4>
%61 = getelementptr inbounds i32, i32* %4, i64 %57
%62 = bitcast i32* %61 to <4 x i32>*
store <4 x i32> %58, <4 x i32>* %62
%63 = getelementptr inbounds i32, i32* %61, i64 4
%64 = bitcast i32* %63 to <4 x i32>*
store <4 x i32> %60, <4 x i32>* %64
%65 = add nuw i64 %57, 8
%66 = add <4 x i32> %58, <i32 8, i32 8, i32 8, i32 8>
%67 = add i64 %59, 1
%68 = icmp eq i64 %67, %14
br i1 %68, label %69, label %56
69: ; preds = %56, %52
%70 = icmp eq i64 %10, %7
br i1 %70, label %73, label %71
71: ; preds = %6, %69
%72 = phi i64 [ 0, %6 ], [ %10, %69 ]
br label %74
73: ; preds = %74, %69, %1
ret i32* %4
74: ; preds = %71, %74
%75 = phi i64 [ %78, %74 ], [ %72, %71 ]
%76 = getelementptr inbounds i32, i32* %4, i64 %75
%77 = trunc i64 %75 to i32
store i32 %77, i32* %76
%78 = add nuw nsw i64 %75, 1
%79 = icmp eq i64 %78, %7
br i1 %79, label %73, label %74
}