add green

Author: IsHYuhi

ABC/ABC124/D.py

@@ -0,0 +1,52 @@
+n, k = map(int, input().split())
+s = list(map(int, list(input())))
+comp = []
+
+c = 1
+p = s[0]
+for i in range(1, n):
+    if s[i] == p:
+        c += 1
+
+    else:
+        comp.append([s[i-1], c])
+        c = 1
+        p = s[i]
+
+comp.append([s[n-1], c])
+
+l = len(comp)
+count = 0
+d = 0
+sum_ = 0
+ans = []
+while d<l:
+    if comp[d][0]==1 and count==k:
+        sum_ += comp[d][1]
+        break
+
+    elif comp[d][0]==0:
+        count += 1
+    sum_ += comp[d][1]
+    d+=1
+
+ans.append(sum_)
+for j in range(d+1, l, 2):
+    if j<l:#明示
+        sum_ += comp[j][1]
+
+    if j+1<l:
+        sum_ += comp[j+1][1]
+
+    if j-k*2-1>=0:
+        sum_ -= comp[j-k*2-1][1]
+
+    if j-k*2>=0:
+        sum_ -= comp[j-k*2][1]
+
+    ans.append(sum_)
+
+if not ans:
+    print(n)
+else:
+    print(max(ans))

ABC/ABC129/D.py

@@ -0,0 +1,37 @@
+#AC with PyPy3, TLE with Python3
+
+#import numpy as np
+h, w = map(int, input().split())
+field = [input() for _ in range(h)]
+acc_field = [[0]*w for _ in range(h)]
+ans = 0
+for i in range(h):
+    to_right = [0]*w
+    to_left = [0]*w
+    to_right[0] = 1 if field[i][0] == '.' else 0
+    to_left[w-1] = 1 if field[i][w-1] == '.' else 0
+
+    for j in range(1, w):
+        to_right[j] = to_right[j-1] + 1 if field[i][j] == '.' else 0
+        to_left[w-j-1] = to_left[w-j] + 1 if field[i][w-j-1] == '.' else 0
+
+    for j in range(w):
+        acc_field[i][j] = max(0, to_right[j]+to_left[j]-1)
+
+for j in range(w):
+    to_lower = [0]*h
+    to_upper = [0]*h
+    to_lower[0] = 1 if field[0][j] == '.' else 0
+    to_upper[h-1] = 1 if field[h-1][j] == '.' else 0
+
+    for i in range(1, h):
+        to_lower[i] = to_lower[i-1] + 1 if field[i][j] == '.' else 0
+        to_upper[h-i-1] = to_upper[h-i] + 1 if field[h-i-1][j] == '.' else 0
+
+    for i in range(h):
+        acc_field[i][j] += max(0, to_lower[i]+to_upper[i]-1)
+        ans = max(ans, acc_field[i][j]-1)
+
+print(ans)
+#print(max(0, np.max(acc_field)-1))
+

ABC/ABC152/D.py

@@ -0,0 +1,23 @@
+n = int(input())
+dic_a = dict()
+dic_b = dict()
+
+for i in range(1, n+1):
+    s = str(i)
+    if dic_a.get((s[0], s[-1])):
+        dic_a[(s[0], s[-1])] += 1
+    else:
+        dic_a[(s[0], s[-1])] = 1
+
+    if dic_b.get((s[-1], s[0])):
+        dic_b[(s[-1], s[0])] += 1
+    else:
+        dic_b[(s[-1], s[0])] = 1
+
+ans = 0
+for key in dic_a.keys():
+    if dic_b.get(key):
+        inter = dic_b[key] * dic_a[key]
+        ans += inter
+
+print(ans)