#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 sqrt(int);
int solve(int dis) {
  int u, steps;
  int remain;
  int i;
  if (dis == 0)
    return 0;
  else if (dis == 1)
    return 1;
  else if (dis == 2)
    return 2;
  u = sqrt(dis);
  steps = 2 * (u - 1);
  remain = dis - u * (u - 1);
  for (i = u; remain != 0; --i)
    if (remain >= i) {
      steps += remain / i;
      ;
      remain %= i;
    }
  return steps;
}


x86 assembly (clang, -O0):
<solve>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %edi,-0x8(%rbp)
cmpl   $0x0,-0x8(%rbp)
jne    1131 <solve+0x21>
movl   $0x0,-0x4(%rbp)
jmp    11e0 <solve+0xd0>
cmpl   $0x1,-0x8(%rbp)
jne    1147 <solve+0x37>
movl   $0x1,-0x4(%rbp)
jmp    11e0 <solve+0xd0>
cmpl   $0x2,-0x8(%rbp)
jne    115d <solve+0x4d>
movl   $0x2,-0x4(%rbp)
jmp    11e0 <solve+0xd0>
jmp    1162 <solve+0x52>
jmp    1167 <solve+0x57>
mov    -0x8(%rbp),%edi
call   1030 <sqrt@plt>
mov    %eax,-0xc(%rbp)
mov    -0xc(%rbp),%eax
sub    $0x1,%eax
shl    $0x1,%eax
mov    %eax,-0x10(%rbp)
mov    -0x8(%rbp),%eax
mov    -0xc(%rbp),%ecx
mov    -0xc(%rbp),%edx
sub    $0x1,%edx
imul   %edx,%ecx
sub    %ecx,%eax
mov    %eax,-0x14(%rbp)
mov    -0xc(%rbp),%eax
mov    %eax,-0x18(%rbp)
cmpl   $0x0,-0x14(%rbp)
je     11da <solve+0xca>
mov    -0x14(%rbp),%eax
cmp    -0x18(%rbp),%eax
jl     11c7 <solve+0xb7>
mov    -0x14(%rbp),%eax
cltd
idivl  -0x18(%rbp)
add    -0x10(%rbp),%eax
mov    %eax,-0x10(%rbp)
mov    -0x18(%rbp),%ecx
mov    -0x14(%rbp),%eax
cltd
idiv   %ecx
mov    %edx,-0x14(%rbp)
jmp    11cc <solve+0xbc>
mov    -0x18(%rbp),%eax
add    $0xffffffff,%eax
mov    %eax,-0x18(%rbp)
jmp    1198 <solve+0x88>
mov    -0x10(%rbp),%eax
mov    %eax,-0x4(%rbp)
mov    -0x4(%rbp),%eax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<solve>:
push   %rbx
mov    %edi,%ebx
cmp    $0x3,%edi
jae    111c <solve+0xc>
mov    %ebx,%ecx
jmp    1135 <solve+0x25>
mov    %ebx,%edi
call   1030 <sqrt@plt>
mov    %eax,%esi
lea    -0x1(%rsi),%eax
lea    (%rsi,%rsi,1),%ecx
add    $0xfffffffe,%ecx
imul   %esi,%eax
sub    %eax,%ebx
jne    1147 <solve+0x37>
mov    %ecx,%eax
pop    %rbx
ret
nopl   0x0(%rax)
add    $0xffffffff,%esi
test   %ebx,%ebx
je     1135 <solve+0x25>
cmp    %esi,%ebx
jl     1140 <solve+0x30>
mov    %ebx,%eax
cltd
idiv   %esi
add    %eax,%ecx
mov    %edx,%ebx
jmp    1140 <solve+0x30>

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

LLVM IR (-O2):
define i32 @solve(i32 %0) {
%2 = icmp ult i32 %0, 3
br i1 %2, label %24, label %3
3: ; preds = %1
%4 = tail call i32 @sqrt(i32 %0)
%5 = add nsw i32 %4, -1
%6 = shl nsw i32 %5, 1
%7 = mul nsw i32 %5, %4
%8 = sub nsw i32 %0, %7
%9 = icmp eq i32 %8, 0
br i1 %9, label %24, label %10
10: ; preds = %3, %19
%11 = phi i32 [ %22, %19 ], [ %4, %3 ]
%12 = phi i32 [ %21, %19 ], [ %8, %3 ]
%13 = phi i32 [ %20, %19 ], [ %6, %3 ]
%14 = icmp slt i32 %12, %11
br i1 %14, label %19, label %15
15: ; preds = %10
%16 = sdiv i32 %12, %11
%17 = add nsw i32 %16, %13
%18 = srem i32 %12, %11
br label %19
19: ; preds = %10, %15
%20 = phi i32 [ %17, %15 ], [ %13, %10 ]
%21 = phi i32 [ %18, %15 ], [ %12, %10 ]
%22 = add nsw i32 %11, -1
%23 = icmp eq i32 %21, 0
br i1 %23, label %24, label %10
24: ; preds = %19, %1, %3
%25 = phi i32 [ %6, %3 ], [ %0, %1 ], [ %20, %19 ]
ret i32 %25
}