leetcode刷题——数组（前缀和/滑动窗口）

209.长度最小的子数组

209. 长度最小的子数组

给定一个含有 n 个正整数的数组和一个正整数 target 。
找出该数组中满足其总和大于等于 target 的长度最小的子数组[numsl, numsl+1, …, numsr-1, numsr] ，并返回其长度。如果不存在符合条件的子数组，返回 0 。

一个控制开始位置，一个遍历结束位置，更新最小长度
注意：

• 连续子数组！连续！！
• 一定要维护最小长度，刚开始要找个最大值 float(“inf”)
• while 那里要一直更新哦

 class Solution:
   def minSubArrayLen(self, target: int, nums: List[int]) -> int:
       min_len=float('inf')
       sum=0
       i=0
       for j in range(len(nums)):
           sum=sum+nums[j]
           while sum>=target:#一直判断
               min_len=min(min_len,j-i+1)
               sum=sum-nums[i]
               i+=1
       return 0 if min_len==float("inf") else min_len

59.螺旋矩阵II

59. 螺旋矩阵 II

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

这种要按照顺序来，先从左到右，然后从上到下，然后从右到左，从下到上
先确定好四个点，最左，最右，最上，最下
左闭右开(left,right+1) （top,bottom+1) (right,left-1,-1) (bottom,top-1,-1)
记得遍历一次 更新 left,right,top,bottom
注意：

• 生成二维矩阵的写法：res=[ [0]*n for _ in range(n)]
• while的条件 top<=bottom and left<=right

 class Solution:
   def generateMatrix(self, n: int) -> List[List[int]]:
       res=[ [0]*n for _ in range(n)]
       left,right,top,bottom=0,n-1,0,n-1
       num=1
       while top<=bottom and left<=right:
           for i in range(left,right+1):
               res[top][i]=num
               num+=1
           top+=1
           for i in range(top,bottom+1):
               res[i][right]=num
               num+=1
           right-=1
           for i in range(right,left-1,-1):
               res[bottom][i]=num
               num+=1
           bottom-=1
           for i in range(bottom,top-1,-1):
               res[i][left]=num
               num+=1
           left+=1
       return res

区间和

https://kamacoder.com/problempage.php?pid=1070

题目描述
给定一个整数数组 Array，请计算该数组在每个指定区间内元素的总和。
输入描述
第一行输入为整数数组 Array 的长度 n，接下来 n 行，每行一个整数，表示数组的元素。随后的输入为需要计算总和的区间下标：a，b （b > = a），直至文件结束。
输出描述
输出每个指定区间内元素的总和。

注意：
acm模式，难点在于处理不定行输入，用sys.stdin.read, data=input().split()
最后要判断 a是否等于0，等于的话就直接qianzhui[b]

import sys
input = sys.stdin.read
data=input().split()
#['2','1','6'] 这种形式
n=int(data[0])
res=[]
for i in range(1,n+1):
    res.append(int(data[i]))
index=n+1
qianzhui=[0]*n
qianzhui[0]=res[0] 
for i in range(1,n):
    qianzhui[i]=qianzhui[i-1]+res[i]

while index<len(data):
    a,b=int(data[index]),int(data[index+1])
    if a==0:
        print(qianzhui[b])
    else:
        print(qianzhui[b]-qianzhui[a-1])
    index=index+2

开发商购买土地

https://kamacoder.com/problempage.php?pid=1044

在一个城市区域内，被划分成了n * m个连续的区块，每个区块都拥有不同的权值，代表着其土地价值。目前，有两家开发公司，A 公司和 B 公司，希望购买这个城市区域的土地。
现在，需要将这个城市区域的所有区块分配给 A 公司和 B 公司。
然而，由于城市规划的限制，只允许将区域按横向或纵向划分成两个子区域，而且每个子区域都必须包含一个或多个区块。
为了确保公平竞争，你需要找到一种分配方式，使得 A 公司和 B 公司各自的子区域内的土地总价值之差最小。
注意：区块不可再分。

注意：
划分的方式只能横着或者竖着
按列/按行找前缀和 就能统计出来

def main():
    import sys
    input = sys.stdin.read
    data = input().split()

    idx = 0
    n = int(data[idx])
    idx += 1
    m = int(data[idx])
    idx += 1
    sum = 0
    vec = []
    for i in range(n):
        row = []
        for j in range(m):
            num = int(data[idx])
            idx += 1
            row.append(num)
            sum += num
        vec.append(row)

    # 统计横向
    horizontal = [0] * n
    for i in range(n):
        for j in range(m):
            horizontal[i] += vec[i][j]

    # 统计纵向
    vertical = [0] * m
    for j in range(m):
        for i in range(n):
            vertical[j] += vec[i][j]

    result = float('inf')
    horizontalCut = 0
    for i in range(n):
        horizontalCut += horizontal[i]
        result = min(result, abs(sum - 2 * horizontalCut))

    verticalCut = 0
    for j in range(m):
        verticalCut += vertical[j]
        result = min(result, abs(sum - 2 * verticalCut))

    print(result)

if __name__ == "__main__":
    main()