#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 */
int /*<<< orphan*/ assert(int);
int /*<<< orphan*/ exist_in_list(int, int *, int);
scalar_t__ malloc(int);
int *get_test_ids(int k, int k_train, int *train_ids) {
  int k_test = k - k_train;
  int *test_ids = (int *)malloc(k_test * sizeof(int));
  int i;
  for (i = 0; i < k_test; i++)
    test_ids[i] = -1;
  int t = 0;
  for (i = 0; i < k; i++) {
    if (!exist_in_list(i, train_ids, k_train)) {
      test_ids[t] = i;
      t++;
    }
  }
  for (i = 0; i < k_test; i++)
    assert(test_ids[i] >= 0);
  return test_ids;
}


x86 assembly (clang, -O0):
<get_test_ids>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %edi,-0x4(%rbp)
mov    %esi,-0x8(%rbp)
mov    %rdx,-0x10(%rbp)
mov    -0x4(%rbp),%eax
sub    -0x8(%rbp),%eax
mov    %eax,-0x14(%rbp)
movslq -0x14(%rbp),%rax
shl    $0x2,%rax
mov    %eax,%edi
call   1040 <malloc@plt>
mov    %rax,-0x20(%rbp)
movl   $0x0,-0x24(%rbp)
mov    -0x24(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    118e <get_test_ids+0x5e>
mov    -0x20(%rbp),%rax
movslq -0x24(%rbp),%rcx
movl   $0xffffffff,(%rax,%rcx,4)
mov    -0x24(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x24(%rbp)
jmp    1165 <get_test_ids+0x35>
movl   $0x0,-0x28(%rbp)
movl   $0x0,-0x24(%rbp)
mov    -0x24(%rbp),%eax
cmp    -0x4(%rbp),%eax
jge    11ea <get_test_ids+0xba>
mov    -0x24(%rbp),%edi
mov    -0x10(%rbp),%rsi
mov    -0x8(%rbp),%edx
call   1050 <exist_in_list@plt>
cmp    $0x0,%eax
jne    11d7 <get_test_ids+0xa7>
mov    -0x24(%rbp),%edx
mov    -0x20(%rbp),%rax
movslq -0x28(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x28(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x28(%rbp)
jmp    11dc <get_test_ids+0xac>
mov    -0x24(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x24(%rbp)
jmp    119c <get_test_ids+0x6c>
movl   $0x0,-0x24(%rbp)
mov    -0x24(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    1224 <get_test_ids+0xf4>
mov    -0x20(%rbp),%rax
movslq -0x24(%rbp),%rcx
cmpl   $0x0,(%rax,%rcx,4)
setge  %al
and    $0x1,%al
movzbl %al,%edi
call   1030 <assert@plt>
mov    -0x24(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x24(%rbp)
jmp    11f1 <get_test_ids+0xc1>
mov    -0x20(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_test_ids>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
push   %rax
mov    %rdx,%rbx
mov    %esi,%r15d
mov    %edi,%r12d
mov    %edi,%ebp
sub    %esi,%ebp
lea    0x0(,%rbp,4),%edi
call   1050 <malloc@plt>
mov    %rax,%r14
mov    %rbp,(%rsp)
test   %ebp,%ebp
jle    118c <get_test_ids+0x4c>
mov    %r15d,%eax
not    %eax
add    %r12d,%eax
lea    0x4(,%rax,4),%rdx
mov    %r14,%rdi
mov    $0xff,%esi
call   1030 <memset@plt>
test   %r12d,%r12d
jle    11c6 <get_test_ids+0x86>
xor    %r13d,%r13d
xor    %ebp,%ebp
jmp    11a8 <get_test_ids+0x68>
nopl   0x0(%rax,%rax,1)
add    $0x1,%ebp
cmp    %ebp,%r12d
je     11c6 <get_test_ids+0x86>
mov    %ebp,%edi
mov    %rbx,%rsi
mov    %r15d,%edx
call   1060 <exist_in_list@plt>
test   %eax,%eax
jne    11a0 <get_test_ids+0x60>
movslq %r13d,%r13
mov    %ebp,(%r14,%r13,4)
add    $0x1,%r13d
jmp    11a0 <get_test_ids+0x60>
mov    (%rsp),%rax
test   %eax,%eax
jle    11f7 <get_test_ids+0xb7>
mov    %eax,%ebp
xor    %ebx,%ebx
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax)
mov    (%r14,%rbx,4),%edi
not    %edi
shr    $0x1f,%edi
call   1040 <assert@plt>
add    $0x1,%rbx
cmp    %rbx,%rbp
jne    11e0 <get_test_ids+0xa0>
mov    %r14,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

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

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