#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 */
unsigned int isqrt(unsigned int);
unsigned int compute_triangle_count(unsigned int count,
                                    unsigned int nrepcodes) {
  unsigned int perrep;
  unsigned int cost;
  cost = 0;
  perrep = (nrepcodes * (nrepcodes + 1)) / 2;
  while (count >= perrep) {
    cost += nrepcodes;
    count -= perrep;
  }
  if (count > 0) {
    unsigned int n;
    n = isqrt(count);
    while ((n * (n + 1)) >= 2 * count)
      n--;
    while ((n * (n + 1)) < 2 * count)
      n++;
    cost += n;
  }
  return (cost);
}


x86 assembly (clang, -O0):
<compute_triangle_count>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %edi,-0x4(%rbp)
mov    %esi,-0x8(%rbp)
movl   $0x0,-0x10(%rbp)
mov    -0x8(%rbp),%eax
mov    -0x8(%rbp),%ecx
add    $0x1,%ecx
imul   %ecx,%eax
shr    $0x1,%eax
mov    %eax,-0xc(%rbp)
mov    -0x4(%rbp),%eax
cmp    -0xc(%rbp),%eax
jb     115c <compute_triangle_count+0x4c>
mov    -0x8(%rbp),%eax
add    -0x10(%rbp),%eax
mov    %eax,-0x10(%rbp)
mov    -0xc(%rbp),%ecx
mov    -0x4(%rbp),%eax
sub    %ecx,%eax
mov    %eax,-0x4(%rbp)
jmp    1137 <compute_triangle_count+0x27>
cmpl   $0x0,-0x4(%rbp)
jbe    11cf <compute_triangle_count+0xbf>
mov    -0x4(%rbp),%edi
call   1030 <isqrt@plt>
mov    %eax,-0x14(%rbp)
mov    -0x14(%rbp),%eax
mov    -0x14(%rbp),%ecx
add    $0x1,%ecx
imul   %ecx,%eax
mov    -0x4(%rbp),%ecx
shl    $0x1,%ecx
cmp    %ecx,%eax
jb     1199 <compute_triangle_count+0x89>
mov    -0x14(%rbp),%eax
add    $0xffffffff,%eax
mov    %eax,-0x14(%rbp)
jmp    1171 <compute_triangle_count+0x61>
jmp    119e <compute_triangle_count+0x8e>
mov    -0x14(%rbp),%eax
mov    -0x14(%rbp),%ecx
add    $0x1,%ecx
imul   %ecx,%eax
mov    -0x4(%rbp),%ecx
shl    $0x1,%ecx
cmp    %ecx,%eax
jae    11c6 <compute_triangle_count+0xb6>
mov    -0x14(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x14(%rbp)
jmp    119e <compute_triangle_count+0x8e>
mov    -0x14(%rbp),%eax
add    -0x10(%rbp),%eax
mov    %eax,-0x10(%rbp)
mov    -0x10(%rbp),%eax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<compute_triangle_count>:
push   %rbp
push   %rbx
push   %rax
mov    %edi,%ebx
lea    0x1(%rsi),%eax
imul   %esi,%eax
shr    %eax
xor    %ebp,%ebp
cmp    %edi,%eax
ja     1138 <compute_triangle_count+0x28>
xor    %ebp,%ebp
cs nopw 0x0(%rax,%rax,1)
nop
add    %esi,%ebp
sub    %eax,%ebx
cmp    %eax,%ebx
jae    1130 <compute_triangle_count+0x20>
test   %ebx,%ebx
je     1180 <compute_triangle_count+0x70>
mov    %ebx,%edi
call   1030 <isqrt@plt>
add    %ebx,%ebx
cs nopw 0x0(%rax,%rax,1)
nop
lea    0x1(%rax),%ecx
imul   %eax,%ecx
add    $0xffffffff,%eax
cmp    %ebx,%ecx
jae    1150 <compute_triangle_count+0x40>
add    %eax,%ebp
add    $0x2,%eax
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax)
lea    -0x1(%rax),%ecx
imul   %eax,%ecx
add    $0x1,%ebp
add    $0x1,%eax
cmp    %ebx,%ecx
jb     1170 <compute_triangle_count+0x60>
mov    %ebp,%eax
add    $0x8,%rsp
pop    %rbx
pop    %rbp
ret

LLVM IR (-O0):
define i32 @compute_triangle_count(i32 %0, i32 %1) {
%3 = alloca i32
%4 = alloca i32
%5 = alloca i32
%6 = alloca i32
%7 = alloca i32
store i32 %0, i32* %3
store i32 %1, i32* %4
store i32 0, i32* %6
%8 = load i32, i32* %4
%9 = load i32, i32* %4
%10 = add i32 %9, 1
%11 = mul i32 %8, %10
%12 = udiv i32 %11, 2
store i32 %12, i32* %5
br label %13
13: ; preds = %17, %2
%14 = load i32, i32* %3
%15 = load i32, i32* %5
%16 = icmp uge i32 %14, %15
br i1 %16, label %17, label %24
17: ; preds = %13
%18 = load i32, i32* %4
%19 = load i32, i32* %6
%20 = add i32 %19, %18
store i32 %20, i32* %6
%21 = load i32, i32* %5
%22 = load i32, i32* %3
%23 = sub i32 %22, %21
store i32 %23, i32* %3
br label %13
24: ; preds = %13
%25 = load i32, i32* %3
%26 = icmp ugt i32 %25, 0
br i1 %26, label %27, label %57
27: ; preds = %24
%28 = load i32, i32* %3
%29 = call i32 @isqrt(i32 %28)
store i32 %29, i32* %7
br label %30
30: ; preds = %38, %27
%31 = load i32, i32* %7
%32 = load i32, i32* %7
%33 = add i32 %32, 1
%34 = mul i32 %31, %33
%35 = load i32, i32* %3
%36 = mul i32 2, %35
%37 = icmp uge i32 %34, %36
br i1 %37, label %38, label %41
38: ; preds = %30
%39 = load i32, i32* %7
%40 = add i32 %39, -1
store i32 %40, i32* %7
br label %30
41: ; preds = %30
br label %42
42: ; preds = %50, %41
%43 = load i32, i32* %7
%44 = load i32, i32* %7
%45 = add i32 %44, 1
%46 = mul i32 %43, %45
%47 = load i32, i32* %3
%48 = mul i32 2, %47
%49 = icmp ult i32 %46, %48
br i1 %49, label %50, label %53
50: ; preds = %42
%51 = load i32, i32* %7
%52 = add i32 %51, 1
store i32 %52, i32* %7
br label %42
53: ; preds = %42
%54 = load i32, i32* %7
%55 = load i32, i32* %6
%56 = add i32 %55, %54
store i32 %56, i32* %6
br label %57
57: ; preds = %53, %24
%58 = load i32, i32* %6
ret i32 %58
}

LLVM IR (-O2):
define i32 @compute_triangle_count(i32 %0, i32 %1) {
%3 = add i32 %1, 1
%4 = mul i32 %3, %1
%5 = lshr i32 %4, 1
%6 = icmp ugt i32 %5, %0
br i1 %6, label %13, label %7
7: ; preds = %2, %7
%8 = phi i32 [ %10, %7 ], [ 0, %2 ]
%9 = phi i32 [ %11, %7 ], [ %0, %2 ]
%10 = add i32 %8, %1
%11 = sub i32 %9, %5
%12 = icmp ult i32 %11, %5
br i1 %12, label %13, label %7
13: ; preds = %7, %2
%14 = phi i32 [ %0, %2 ], [ %11, %7 ]
%15 = phi i32 [ 0, %2 ], [ %10, %7 ]
%16 = icmp eq i32 %14, 0
br i1 %16, label %33, label %17
17: ; preds = %13
%18 = tail call i32 @isqrt(i32 %14)
%19 = shl i32 %14, 1
br label %20
20: ; preds = %20, %17
%21 = phi i32 [ %18, %17 ], [ %25, %20 ]
%22 = add i32 %21, 1
%23 = mul i32 %22, %21
%24 = icmp ult i32 %23, %19
%25 = add i32 %21, -1
br i1 %24, label %26, label %20
26: ; preds = %20, %26
%27 = phi i32 [ %28, %26 ], [ %21, %20 ]
%28 = add i32 %27, 1
%29 = mul i32 %28, %27
%30 = icmp ult i32 %29, %19
br i1 %30, label %26, label %31
31: ; preds = %26
%32 = add i32 %27, %15
br label %33
33: ; preds = %31, %13
%34 = phi i32 [ %32, %31 ], [ %15, %13 ]
ret i32 %34
}