-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest.py
90 lines (63 loc) · 1.54 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import numpy as np
s = open("inputs22/8").read()
t = [[int(j) for j in i] for i in s.split("\n")]
print(t)
F = np.array(t)
print(F)
v = 0
N, M = F.shape
for i in range(N):
for j in range(M):
val = F[i,j]
if ((val > F[:i,j]).all()
or (val > F[i+1:,j]).all()
or (val > F[i,:j]).all()
or (val > F[i,j+1:]).all()):
v += 1
print(v)
def compute_ssc(slice):
if slice.shape[0] <= 1: return 1
tree = slice[0]
ss = 1
ret = None
for next_tree in slice[1:]:
print(next_tree)
if next_tree >= tree:
ret = ss
else:
ss += 1
if not ret: ret = ss-1
print(slice, ret)
return ret
# print(slice)
# t = slice[0]
# ss = 1
# pi = slice[1]
# if pi > t: return 1
# for i in slice[2:]:
# if i <= t:
# pi = i
# ss += 1
# else:
# break
# return ss
SF = np.zeros_like(F)
for i in range(N):
for j in range(M):
# i = 1
# j = 2
# if i != 2 or j != 0: continue
val = F[i,j]
up = compute_ssc(F[:i+1,j][::-1])
left = compute_ssc(F[i,:j+1][::-1])
down = compute_ssc(F[i:,j])
right = compute_ssc(F[i,j:])
print(val, (i,j), "==>", up, down, left, right)
SF[i,j] = up * down * left * right
print(SF)
print(np.max(SF[1:-1,1:-1]))
# import pandas as pd
# dic = {("k1", "k2"): 0.1, ("k21", "k22"): 0.2}
# data = [[k1, k2, k3] for (k1, k2), k3 in dic.items()]
# print(data)
# print(pd.DataFrame(data))