forked from NUDT-compiler/nudt-compiler-cpp
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
495 lines
8.4 KiB
495 lines
8.4 KiB
.data
|
|
.globl INF
|
|
.p2align 2
|
|
INF:
|
|
.word 1879048192
|
|
.globl size
|
|
.p2align 2
|
|
size:
|
|
.zero 40
|
|
.globl to
|
|
.p2align 2
|
|
to:
|
|
.zero 400
|
|
.globl cap
|
|
.p2align 2
|
|
cap:
|
|
.zero 400
|
|
.globl rev
|
|
.p2align 2
|
|
rev:
|
|
.zero 400
|
|
.globl used
|
|
.p2align 2
|
|
used:
|
|
.zero 40
|
|
|
|
.text
|
|
.globl my_memset
|
|
.p2align 2
|
|
my_memset:
|
|
.L.my_memset.0:
|
|
stp x29, x30, [sp, #-16]!
|
|
mov x29, sp
|
|
mov x8, x0
|
|
mov w9, w1
|
|
mov w10, w2
|
|
mov w11, #0
|
|
b .L.my_memset.1
|
|
.L.my_memset.1:
|
|
cmp w11, w10
|
|
b.lt .L.my_memset.2
|
|
b .L.my_memset.3
|
|
.L.my_memset.2:
|
|
sxtw x12, w11
|
|
lsl x12, x12, #2
|
|
add x12, x8, x12
|
|
str w9, [x12]
|
|
mov w12, #1
|
|
add w11, w11, w12
|
|
b .L.my_memset.1
|
|
.L.my_memset.3:
|
|
ldp x29, x30, [sp], #16
|
|
ret
|
|
|
|
.text
|
|
.globl add_node
|
|
.p2align 2
|
|
add_node:
|
|
.L.add_node.0:
|
|
stp x29, x30, [sp, #-16]!
|
|
mov x29, sp
|
|
sub sp, sp, #48
|
|
str x21, [sp, #0]
|
|
str x19, [sp, #8]
|
|
str x20, [sp, #16]
|
|
mov w10, w0
|
|
mov w11, w1
|
|
mov w21, w2
|
|
adrp x12, size
|
|
add x12, x12, :lo12:size
|
|
sxtw x8, w10
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
mov w19, #10
|
|
mul w15, w10, w19
|
|
add w8, w15, w8
|
|
adrp x20, to
|
|
add x20, x20, :lo12:to
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x20, x8
|
|
str w11, [x8]
|
|
sxtw x8, w10
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
add w8, w15, w8
|
|
adrp x9, cap
|
|
add x9, x9, :lo12:cap
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x9, x8
|
|
str w21, [x8]
|
|
sxtw x8, w10
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
add w8, w15, w8
|
|
sxtw x15, w11
|
|
lsl x15, x15, #2
|
|
add x15, x12, x15
|
|
ldr w21, [x15]
|
|
adrp x15, rev
|
|
add x15, x15, :lo12:rev
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x15, x8
|
|
str w21, [x8]
|
|
sxtw x8, w11
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
mul w19, w11, w19
|
|
add w8, w19, w8
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x20, x8
|
|
str w10, [x8]
|
|
sxtw x8, w11
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
add w8, w19, w8
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x9, x9, x8
|
|
mov w8, #0
|
|
str w8, [x9]
|
|
sxtw x8, w11
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
add w8, w19, w8
|
|
sxtw x9, w10
|
|
lsl x9, x9, #2
|
|
add x9, x12, x9
|
|
ldr w9, [x9]
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x15, x8
|
|
str w9, [x8]
|
|
sxtw x8, w10
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
mov w15, #1
|
|
add w9, w8, w15
|
|
sxtw x8, w10
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
str w9, [x8]
|
|
sxtw x8, w11
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
add w9, w8, w15
|
|
sxtw x8, w11
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
str w9, [x8]
|
|
ldr x21, [sp, #0]
|
|
ldr x19, [sp, #8]
|
|
ldr x20, [sp, #16]
|
|
add sp, sp, #48
|
|
ldp x29, x30, [sp], #16
|
|
ret
|
|
|
|
.text
|
|
.globl dfs
|
|
.p2align 2
|
|
dfs:
|
|
.L.dfs.0:
|
|
stp x29, x30, [sp, #-16]!
|
|
mov x29, sp
|
|
sub sp, sp, #48
|
|
str x21, [sp, #0]
|
|
str x20, [sp, #8]
|
|
str x19, [sp, #16]
|
|
str x22, [sp, #24]
|
|
mov w20, w0
|
|
mov w22, w1
|
|
mov w21, w2
|
|
cmp w20, w22
|
|
b.eq .L.dfs.1
|
|
b .L.dfs.2
|
|
.L.dfs.1:
|
|
mov w0, w21
|
|
ldr x21, [sp, #0]
|
|
ldr x20, [sp, #8]
|
|
ldr x19, [sp, #16]
|
|
ldr x22, [sp, #24]
|
|
add sp, sp, #48
|
|
ldp x29, x30, [sp], #16
|
|
ret
|
|
.L.dfs.2:
|
|
adrp x9, used
|
|
add x9, x9, :lo12:used
|
|
sxtw x8, w20
|
|
lsl x8, x8, #2
|
|
add x9, x9, x8
|
|
mov w8, #1
|
|
str w8, [x9]
|
|
mov w8, #0
|
|
mov w19, w8
|
|
b .L.dfs.3
|
|
.L.dfs.3:
|
|
adrp x9, size
|
|
add x9, x9, :lo12:size
|
|
sxtw x8, w20
|
|
lsl x8, x8, #2
|
|
add x8, x9, x8
|
|
ldr w8, [x8]
|
|
cmp w19, w8
|
|
b.lt .L.dfs.4
|
|
b .L.dfs.5
|
|
.L.dfs.4:
|
|
mov w8, #10
|
|
mul w8, w20, w8
|
|
add w8, w8, w19
|
|
adrp x9, to
|
|
add x9, x9, :lo12:to
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x9, x8
|
|
ldr w8, [x8]
|
|
adrp x9, used
|
|
add x9, x9, :lo12:used
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x9, x8
|
|
ldr w9, [x8]
|
|
mov w8, #0
|
|
cmp w9, w8
|
|
b.ne .L.dfs.6
|
|
b .L.dfs.7
|
|
.L.dfs.5:
|
|
mov w8, #0
|
|
mov w0, w8
|
|
ldr x21, [sp, #0]
|
|
ldr x20, [sp, #8]
|
|
ldr x19, [sp, #16]
|
|
ldr x22, [sp, #24]
|
|
add sp, sp, #48
|
|
ldp x29, x30, [sp], #16
|
|
ret
|
|
.L.dfs.6:
|
|
mov w8, #1
|
|
add w8, w19, w8
|
|
mov w19, w8
|
|
b .L.dfs.3
|
|
.L.dfs.7:
|
|
mov w8, #10
|
|
mul w8, w20, w8
|
|
add w8, w8, w19
|
|
adrp x9, cap
|
|
add x9, x9, :lo12:cap
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x9, x8
|
|
ldr w9, [x8]
|
|
mov w8, #0
|
|
cmp w9, w8
|
|
b.le .L.dfs.8
|
|
b .L.dfs.9
|
|
.L.dfs.8:
|
|
mov w8, #1
|
|
add w8, w19, w8
|
|
mov w19, w8
|
|
b .L.dfs.3
|
|
.L.dfs.9:
|
|
mov w8, #10
|
|
mul w8, w20, w8
|
|
add w8, w8, w19
|
|
adrp x9, cap
|
|
add x9, x9, :lo12:cap
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x9, x8
|
|
ldr w8, [x8]
|
|
cmp w21, w8
|
|
b.lt .L.dfs.10
|
|
b .L.dfs.11
|
|
.L.dfs.10:
|
|
mov w8, w21
|
|
b .L.dfs.12
|
|
.L.dfs.11:
|
|
mov w8, #10
|
|
mul w8, w20, w8
|
|
add w8, w8, w19
|
|
adrp x9, cap
|
|
add x9, x9, :lo12:cap
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x9, x8
|
|
ldr w8, [x8]
|
|
b .L.dfs.12
|
|
.L.dfs.12:
|
|
mov w9, #10
|
|
mul w9, w20, w9
|
|
add w9, w9, w19
|
|
adrp x10, to
|
|
add x10, x10, :lo12:to
|
|
sxtw x9, w9
|
|
lsl x9, x9, #2
|
|
add x9, x10, x9
|
|
ldr w9, [x9]
|
|
mov w0, w9
|
|
mov w1, w22
|
|
mov w2, w8
|
|
bl dfs
|
|
mov w10, w0
|
|
mov w8, #0
|
|
cmp w10, w8
|
|
b.gt .L.dfs.13
|
|
b .L.dfs.14
|
|
.L.dfs.13:
|
|
mov w11, #10
|
|
mul w8, w20, w11
|
|
add w19, w8, w19
|
|
adrp x12, cap
|
|
add x12, x12, :lo12:cap
|
|
sxtw x8, w19
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
sub w9, w8, w10
|
|
sxtw x8, w19
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
str w9, [x8]
|
|
adrp x21, to
|
|
add x21, x21, :lo12:to
|
|
sxtw x8, w19
|
|
lsl x8, x8, #2
|
|
add x8, x21, x8
|
|
ldr w8, [x8]
|
|
adrp x15, rev
|
|
add x15, x15, :lo12:rev
|
|
sxtw x9, w19
|
|
lsl x9, x9, #2
|
|
add x9, x15, x9
|
|
ldr w9, [x9]
|
|
mul w8, w8, w11
|
|
add w20, w8, w9
|
|
sxtw x8, w19
|
|
lsl x8, x8, #2
|
|
add x8, x21, x8
|
|
ldr w8, [x8]
|
|
sxtw x9, w19
|
|
lsl x9, x9, #2
|
|
add x9, x15, x9
|
|
ldr w9, [x9]
|
|
mul w8, w8, w11
|
|
add w8, w8, w9
|
|
sxtw x8, w8
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
ldr w8, [x8]
|
|
add w9, w8, w10
|
|
sxtw x8, w20
|
|
lsl x8, x8, #2
|
|
add x8, x12, x8
|
|
str w9, [x8]
|
|
mov w0, w10
|
|
ldr x21, [sp, #0]
|
|
ldr x20, [sp, #8]
|
|
ldr x19, [sp, #16]
|
|
ldr x22, [sp, #24]
|
|
add sp, sp, #48
|
|
ldp x29, x30, [sp], #16
|
|
ret
|
|
.L.dfs.14:
|
|
mov w8, #1
|
|
add w8, w19, w8
|
|
mov w19, w8
|
|
b .L.dfs.3
|
|
|
|
.text
|
|
.globl max_flow
|
|
.p2align 2
|
|
max_flow:
|
|
.L.max_flow.0:
|
|
stp x29, x30, [sp, #-16]!
|
|
mov x29, sp
|
|
sub sp, sp, #48
|
|
str x21, [sp, #0]
|
|
str x20, [sp, #8]
|
|
str x19, [sp, #16]
|
|
mov w21, w0
|
|
mov w20, w1
|
|
mov w8, #0
|
|
mov w19, w8
|
|
b .L.max_flow.1
|
|
.L.max_flow.1:
|
|
b .L.max_flow.2
|
|
.L.max_flow.2:
|
|
adrp x8, used
|
|
add x8, x8, :lo12:used
|
|
mov x0, x8
|
|
mov w8, #0
|
|
mov w1, w8
|
|
mov w8, #10
|
|
mov w2, w8
|
|
bl my_memset
|
|
mov w0, w21
|
|
mov w1, w20
|
|
movz w8, #0
|
|
movk w8, #28672, lsl #16
|
|
mov w2, w8
|
|
bl dfs
|
|
mov w9, w0
|
|
mov w8, #0
|
|
cmp w9, w8
|
|
b.eq .L.max_flow.3
|
|
b .L.max_flow.4
|
|
.L.max_flow.3:
|
|
mov w0, w19
|
|
ldr x21, [sp, #0]
|
|
ldr x20, [sp, #8]
|
|
ldr x19, [sp, #16]
|
|
add sp, sp, #48
|
|
ldp x29, x30, [sp], #16
|
|
ret
|
|
.L.max_flow.4:
|
|
add w8, w19, w9
|
|
mov w19, w8
|
|
b .L.max_flow.1
|
|
|
|
.text
|
|
.globl main
|
|
.p2align 2
|
|
main:
|
|
.L.main.0:
|
|
stp x29, x30, [sp, #-16]!
|
|
mov x29, sp
|
|
sub sp, sp, #48
|
|
str x22, [sp, #0]
|
|
str x21, [sp, #8]
|
|
str x19, [sp, #16]
|
|
str x20, [sp, #24]
|
|
bl getint
|
|
mov w20, w0
|
|
bl getint
|
|
mov w19, w0
|
|
adrp x8, size
|
|
add x8, x8, :lo12:size
|
|
mov x0, x8
|
|
mov w8, #0
|
|
mov w1, w8
|
|
mov w8, #10
|
|
mov w2, w8
|
|
bl my_memset
|
|
b .L.main.1
|
|
.L.main.1:
|
|
mov w8, #0
|
|
cmp w19, w8
|
|
b.gt .L.main.2
|
|
b .L.main.3
|
|
.L.main.2:
|
|
bl getint
|
|
mov w21, w0
|
|
bl getint
|
|
mov w22, w0
|
|
bl getint
|
|
mov w8, w0
|
|
mov w0, w21
|
|
mov w1, w22
|
|
mov w2, w8
|
|
bl add_node
|
|
mov w8, #1
|
|
sub w8, w19, w8
|
|
mov w19, w8
|
|
b .L.main.1
|
|
.L.main.3:
|
|
mov w8, #1
|
|
mov w0, w8
|
|
mov w1, w20
|
|
bl max_flow
|
|
mov w8, w0
|
|
mov w0, w8
|
|
bl putint
|
|
mov w8, #10
|
|
mov w0, w8
|
|
bl putch
|
|
mov w8, #0
|
|
mov w0, w8
|
|
ldr x22, [sp, #0]
|
|
ldr x21, [sp, #8]
|
|
ldr x19, [sp, #16]
|
|
ldr x20, [sp, #24]
|
|
add sp, sp, #48
|
|
ldp x29, x30, [sp], #16
|
|
ret
|