#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);
void partition_restr_from_id(void *p_ptr, int n, int k, int id, int *part,
                             int *num_part) {
  int(*p)[n + 1][n + 1] = (int(*)[n + 1][n + 1]) p_ptr;
  assert(id >= 0 && id < (*p)[n][n]);
  int n_loc = n;
  *num_part = 0;
  int psn = (*p)[n][n] - id - 1;
A:
  (*num_part)++;
  int m = k;
B:
  if ((*p)[n_loc][m] < psn) {
    m++;
    goto B;
  }
  if ((*p)[n_loc][m] > psn) {
  C:
    part[*num_part - 1] = m;
    psn -= (*p)[n_loc][m - 1];
    n_loc -= m;
    if (n_loc < k) {
      return;
    } else {
      goto A;
    }
  } else {
    m++;
  }
  if (m == n_loc) {
    goto C;
  } else {
    goto B;
  }
}


x86 assembly (clang, -O0):
<partition_restr_from_id>:
push   %rbp
mov    %rsp,%rbp
sub    $0x50,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
mov    %edx,-0x10(%rbp)
mov    %ecx,-0x14(%rbp)
mov    %r8,-0x20(%rbp)
mov    %r9,-0x28(%rbp)
mov    -0xc(%rbp),%eax
add    $0x1,%eax
mov    %eax,%eax
mov    %rax,-0x48(%rbp)
mov    -0x8(%rbp),%rax
mov    %rax,-0x30(%rbp)
xor    %eax,%eax
cmpl   $0x0,-0x14(%rbp)
mov    %al,-0x3d(%rbp)
jl     1177 <partition_restr_from_id+0x67>
mov    -0x48(%rbp),%rsi
mov    -0x14(%rbp),%eax
mov    -0x30(%rbp),%rcx
movslq -0xc(%rbp),%rdx
imul   %rsi,%rdx
shl    $0x2,%rdx
add    %rdx,%rcx
movslq -0xc(%rbp),%rdx
cmp    (%rcx,%rdx,4),%eax
setl   %al
mov    %al,-0x3d(%rbp)
mov    -0x3d(%rbp),%al
and    $0x1,%al
movzbl %al,%edi
call   1030 <assert@plt>
mov    -0x48(%rbp),%rdx
mov    -0xc(%rbp),%eax
mov    %eax,-0x34(%rbp)
mov    -0x28(%rbp),%rax
movl   $0x0,(%rax)
mov    -0x30(%rbp),%rax
movslq -0xc(%rbp),%rcx
imul   %rdx,%rcx
shl    $0x2,%rcx
add    %rcx,%rax
movslq -0xc(%rbp),%rcx
mov    (%rax,%rcx,4),%eax
sub    -0x14(%rbp),%eax
sub    $0x1,%eax
mov    %eax,-0x38(%rbp)
mov    -0x28(%rbp),%rax
mov    (%rax),%ecx
add    $0x1,%ecx
mov    %ecx,(%rax)
mov    -0x10(%rbp),%eax
mov    %eax,-0x3c(%rbp)
mov    -0x48(%rbp),%rdx
mov    -0x30(%rbp),%rax
movslq -0x34(%rbp),%rcx
imul   %rdx,%rcx
shl    $0x2,%rcx
add    %rcx,%rax
movslq -0x3c(%rbp),%rcx
mov    (%rax,%rcx,4),%eax
cmp    -0x38(%rbp),%eax
jge    1201 <partition_restr_from_id+0xf1>
mov    -0x3c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x3c(%rbp)
jmp    11cc <partition_restr_from_id+0xbc>
mov    -0x48(%rbp),%rdx
mov    -0x30(%rbp),%rax
movslq -0x34(%rbp),%rcx
imul   %rdx,%rcx
shl    $0x2,%rcx
add    %rcx,%rax
movslq -0x3c(%rbp),%rcx
mov    (%rax,%rcx,4),%eax
cmp    -0x38(%rbp),%eax
jle    1290 <partition_restr_from_id+0x180>
jmp    122d <partition_restr_from_id+0x11d>
mov    -0x48(%rbp),%rdx
mov    -0x3c(%rbp),%esi
mov    -0x20(%rbp),%rax
mov    -0x28(%rbp),%rcx
mov    (%rcx),%ecx
sub    $0x1,%ecx
movslq %ecx,%rcx
mov    %esi,(%rax,%rcx,4)
mov    -0x30(%rbp),%rax
movslq -0x34(%rbp),%rcx
imul   %rdx,%rcx
shl    $0x2,%rcx
add    %rcx,%rax
mov    -0x3c(%rbp),%ecx
sub    $0x1,%ecx
movslq %ecx,%rcx
mov    (%rax,%rcx,4),%ecx
mov    -0x38(%rbp),%eax
sub    %ecx,%eax
mov    %eax,-0x38(%rbp)
mov    -0x3c(%rbp),%ecx
mov    -0x34(%rbp),%eax
sub    %ecx,%eax
mov    %eax,-0x34(%rbp)
mov    -0x34(%rbp),%eax
cmp    -0x10(%rbp),%eax
jge    128b <partition_restr_from_id+0x17b>
add    $0x50,%rsp
pop    %rbp
ret
jmp    11bb <partition_restr_from_id+0xab>
mov    -0x3c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x3c(%rbp)
mov    -0x3c(%rbp),%eax
cmp    -0x34(%rbp),%eax
jne    12aa <partition_restr_from_id+0x19a>
jmp    122d <partition_restr_from_id+0x11d>
jmp    11cc <partition_restr_from_id+0xbc>

x86 assembly (clang, -O2):
<partition_restr_from_id>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
sub    $0x28,%rsp
mov    %r9,%r15
mov    %r8,0x20(%rsp)
mov    %ecx,%ebp
mov    %edx,%r12d
mov    %esi,%ebx
mov    %rdi,%r13
lea    0x1(%rbx),%eax
movslq %esi,%rcx
mov    %rax,0x18(%rsp)
mov    %rax,%rdx
imul   %rcx,%rdx
xor    %r14d,%r14d
mov    $0x0,%edi
test   %ebp,%ebp
js     1162 <partition_restr_from_id+0x52>
lea    0x0(,%rdx,4),%rax
add    %r13,%rax
xor    %edi,%edi
cmp    %ebp,(%rax,%rcx,4)
setg   %dil
mov    %rcx,0x10(%rsp)
mov    %rdx,0x8(%rsp)
call   1030 <assert@plt>
movl   $0x0,(%r15)
mov    0x8(%rsp),%rax
lea    0x0(,%rax,4),%rax
add    %r13,%rax
not    %ebp
mov    0x10(%rsp),%rcx
add    (%rax,%rcx,4),%ebp
mov    0x20(%rsp),%rsi
mov    0x18(%rsp),%rdi
nopl   0x0(%rax)
lea    0x1(%r14),%eax
mov    %eax,(%r15)
movslq %ebx,%rax
imul   %rdi,%rax
lea    0x0(,%rax,4),%rax
add    %r13,%rax
mov    %r12d,%ecx
jmp    11c3 <partition_restr_from_id+0xb3>
xchg   %ax,%ax
add    $0x1,%ecx
movslq %ecx,%rdx
cmp    %ebp,(%rax,%rdx,4)
jl     11c0 <partition_restr_from_id+0xb0>
jg     11d6 <partition_restr_from_id+0xc6>
add    $0x1,%ecx
cmp    %ebx,%ecx
jne    11c3 <partition_restr_from_id+0xb3>
mov    %ebx,%ecx
movslq %r14d,%rdx
mov    %ecx,(%rsi,%rdx,4)
sub    %ecx,%ebx
cmp    %r12d,%ebx
jl     11ef <partition_restr_from_id+0xdf>
movslq %ecx,%rcx
sub    -0x4(%rax,%rcx,4),%ebp
mov    (%r15),%r14d
jmp    11a0 <partition_restr_from_id+0x90>
add    $0x28,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define void @partition_restr_from_id(i8* %0, i32 %1, i32 %2, i32 %3, i32* %4, i32* %5) {
%7 = alloca i8*
%8 = alloca i32
%9 = alloca i32
%10 = alloca i32
%11 = alloca i32*
%12 = alloca i32*
%13 = alloca i32*
%14 = alloca i32
%15 = alloca i32
%16 = alloca i32
store i8* %0, i8** %7
store i32 %1, i32* %8
store i32 %2, i32* %9
store i32 %3, i32* %10
store i32* %4, i32** %11
store i32* %5, i32** %12
%17 = load i32, i32* %8
%18 = add nsw i32 %17, 1
%19 = zext i32 %18 to i64
%20 = load i32, i32* %8
%21 = add nsw i32 %20, 1
%22 = zext i32 %21 to i64
%23 = load i32, i32* %8
%24 = add nsw i32 %23, 1
%25 = zext i32 %24 to i64
%26 = load i32, i32* %8
%27 = add nsw i32 %26, 1
%28 = zext i32 %27 to i64
%29 = load i8*, i8** %7
%30 = bitcast i8* %29 to i32*
store i32* %30, i32** %13
%31 = load i32, i32* %10
%32 = icmp sge i32 %31, 0
br i1 %32, label %33, label %45
33: ; preds = %6
%34 = load i32, i32* %10
%35 = load i32*, i32** %13
%36 = load i32, i32* %8
%37 = sext i32 %36 to i64
%38 = mul nsw i64 %37, %22
%39 = getelementptr inbounds i32, i32* %35, i64 %38
%40 = load i32, i32* %8
%41 = sext i32 %40 to i64
%42 = getelementptr inbounds i32, i32* %39, i64 %41
%43 = load i32, i32* %42
%44 = icmp slt i32 %34, %43
br label %45
45: ; preds = %33, %6
%46 = phi i1 [ false, %6 ], [ %44, %33 ]
%47 = zext i1 %46 to i32
%48 = call i32 @assert(i32 %47)
%49 = load i32, i32* %8
store i32 %49, i32* %14
%50 = load i32*, i32** %12
store i32 0, i32* %50
%51 = load i32*, i32** %13
%52 = load i32, i32* %8
%53 = sext i32 %52 to i64
%54 = mul nsw i64 %53, %22
%55 = getelementptr inbounds i32, i32* %51, i64 %54
%56 = load i32, i32* %8
%57 = sext i32 %56 to i64
%58 = getelementptr inbounds i32, i32* %55, i64 %57
%59 = load i32, i32* %58
%60 = load i32, i32* %10
%61 = sub nsw i32 %59, %60
%62 = sub nsw i32 %61, 1
store i32 %62, i32* %15
br label %63
63: ; preds = %123, %45
%64 = load i32*, i32** %12
%65 = load i32, i32* %64
%66 = add nsw i32 %65, 1
store i32 %66, i32* %64
%67 = load i32, i32* %9
store i32 %67, i32* %16
br label %68
68: ; preds = %132, %80, %63
%69 = load i32*, i32** %13
%70 = load i32, i32* %14
%71 = sext i32 %70 to i64
%72 = mul nsw i64 %71, %22
%73 = getelementptr inbounds i32, i32* %69, i64 %72
%74 = load i32, i32* %16
%75 = sext i32 %74 to i64
%76 = getelementptr inbounds i32, i32* %73, i64 %75
%77 = load i32, i32* %76
%78 = load i32, i32* %15
%79 = icmp slt i32 %77, %78
br i1 %79, label %80, label %83
80: ; preds = %68
%81 = load i32, i32* %16
%82 = add nsw i32 %81, 1
store i32 %82, i32* %16
br label %68
83: ; preds = %68
%84 = load i32*, i32** %13
%85 = load i32, i32* %14
%86 = sext i32 %85 to i64
%87 = mul nsw i64 %86, %22
%88 = getelementptr inbounds i32, i32* %84, i64 %87
%89 = load i32, i32* %16
%90 = sext i32 %89 to i64
%91 = getelementptr inbounds i32, i32* %88, i64 %90
%92 = load i32, i32* %91
%93 = load i32, i32* %15
%94 = icmp sgt i32 %92, %93
br i1 %94, label %95, label %124
95: ; preds = %83
br label %96
96: ; preds = %131, %95
%97 = load i32, i32* %16
%98 = load i32*, i32** %11
%99 = load i32*, i32** %12
%100 = load i32, i32* %99
%101 = sub nsw i32 %100, 1
%102 = sext i32 %101 to i64
%103 = getelementptr inbounds i32, i32* %98, i64 %102
store i32 %97, i32* %103
%104 = load i32*, i32** %13
%105 = load i32, i32* %14
%106 = sext i32 %105 to i64
%107 = mul nsw i64 %106, %22
%108 = getelementptr inbounds i32, i32* %104, i64 %107
%109 = load i32, i32* %16
%110 = sub nsw i32 %109, 1
%111 = sext i32 %110 to i64
%112 = getelementptr inbounds i32, i32* %108, i64 %111
%113 = load i32, i32* %112
%114 = load i32, i32* %15
%115 = sub nsw i32 %114, %113
store i32 %115, i32* %15
%116 = load i32, i32* %16
%117 = load i32, i32* %14
%118 = sub nsw i32 %117, %116
store i32 %118, i32* %14
%119 = load i32, i32* %14
%120 = load i32, i32* %9
%121 = icmp slt i32 %119, %120
br i1 %121, label %122, label %123
122: ; preds = %96
ret void
123: ; preds = %96
br label %63
124: ; preds = %83
%125 = load i32, i32* %16
%126 = add nsw i32 %125, 1
store i32 %126, i32* %16
br label %127
127: ; preds = %124
%128 = load i32, i32* %16
%129 = load i32, i32* %14
%130 = icmp eq i32 %128, %129
br i1 %130, label %131, label %132
131: ; preds = %127
br label %96
132: ; preds = %127
br label %68
}

LLVM IR (-O2):
define void @partition_restr_from_id(i8* nocapture %0, i32 %1, i32 %2, i32 %3, i32* nocapture writeonly %4, i32* nocapture %5) {
%7 = add nsw i32 %1, 1
%8 = zext i32 %7 to i64
%9 = bitcast i8* %0 to i32*
%10 = icmp sgt i32 %3, -1
%11 = sext i32 %1 to i64
%12 = mul nsw i64 %8, %11
br i1 %10, label %13, label %19
13: ; preds = %6
%14 = getelementptr inbounds i32, i32* %9, i64 %12
%15 = getelementptr inbounds i32, i32* %14, i64 %11
%16 = load i32, i32* %15
%17 = icmp sgt i32 %16, %3
%18 = zext i1 %17 to i32
br label %19
19: ; preds = %6, %13
%20 = phi i32 [ %18, %13 ], [ 0, %6 ]
%21 = tail call i32 @assert(i32 %20)
store i32 0, i32* %5
%22 = getelementptr inbounds i32, i32* %9, i64 %12
%23 = getelementptr inbounds i32, i32* %22, i64 %11
%24 = load i32, i32* %23
%25 = xor i32 %3, -1
%26 = add i32 %24, %25
br label %27
27: ; preds = %53, %19
%28 = phi i32 [ 0, %19 ], [ %59, %53 ]
%29 = phi i32 [ %1, %19 ], [ %51, %53 ]
%30 = phi i32 [ %26, %19 ], [ %58, %53 ]
%31 = add nsw i32 %28, 1
store i32 %31, i32* %5
%32 = sext i32 %29 to i64
%33 = mul nsw i64 %32, %8
%34 = getelementptr inbounds i32, i32* %9, i64 %33
br label %35
35: ; preds = %43, %27
%36 = phi i32 [ %2, %27 ], [ %44, %43 ]
%37 = sext i32 %36 to i64
%38 = getelementptr inbounds i32, i32* %34, i64 %37
%39 = load i32, i32* %38
%40 = icmp slt i32 %39, %30
br i1 %40, label %41, label %45
41: ; preds = %35
%42 = add nsw i32 %36, 1
br label %43
43: ; preds = %41, %61
%44 = phi i32 [ %42, %41 ], [ %62, %61 ]
br label %35
45: ; preds = %35
%46 = icmp sgt i32 %39, %30
br i1 %46, label %47, label %61
47: ; preds = %61, %45
%48 = phi i32 [ %36, %45 ], [ %29, %61 ]
%49 = sext i32 %28 to i64
%50 = getelementptr inbounds i32, i32* %4, i64 %49
store i32 %48, i32* %50
%51 = sub nsw i32 %29, %48
%52 = icmp slt i32 %51, %2
br i1 %52, label %60, label %53
53: ; preds = %47
%54 = add nsw i32 %48, -1
%55 = sext i32 %54 to i64
%56 = getelementptr inbounds i32, i32* %34, i64 %55
%57 = load i32, i32* %56
%58 = sub nsw i32 %30, %57
%59 = load i32, i32* %5
br label %27
60: ; preds = %47
ret void
61: ; preds = %45
%62 = add nsw i32 %36, 1
%63 = icmp eq i32 %62, %29
br i1 %63, label %47, label %43
}