有意思的各类算法,思维题目分享

1.统计子矩阵

 思路:二维前缀和超时,下面是前缀和加双指针,对列前缀和,两个玄幻控制行号,双指针控制列的移动

考查:前缀和+双指针

import os
import sys

# 请在此输入您的代码
# 矩阵大小  N × M
n,m,k=map(int,input().split())  #默认用空格隔开
s=[[0 for i in range(m+1)]]   #定义一个数组
for i in range(1,n+1):
  s.append([0]+list(map(int,input().split())))
  for j in range(1,m+1):   
    s[i][j]+=s[i-1][j]   # 列的前缀和
res=0
for i in range(1,n+1):
  for j in range(i,n+1):   #枚举上下边界
    l=1          #双指针,对列用的双指针
    sum=0
    for r in range(1,m+1):    #这部分计算的是区域和
      sum+=s[j][r]-s[i-1][r]   #r列的值+
      while sum>k:
        sum-=s[j][l]-s[i-1][l]
        l+=1
      res+=r-l+1
print(res)

附上二维前缀和代码

import os
import sys

# 请在此输入您的代码

n,m,k = map(int,input().split())
a=[[0] for i in range(n)]
a.insert(0,[0]*(m+1))
for i in range(1,n+1):  # 转换二维矩阵形式,即下标从1开始
  a[i].extend(map(int,input().split()))

s = [[0]*(m+1) for i in range(n+1)]  # 预计算前缀和s[][]
for i in range(1,n+1):
  for j in range(1,m+1):
    s[i][j] = s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j]

ans =0
for i1 in range(1,n+1):
  for i2 in range(i1,n+1):
    for j1 in range(1,m+1):
      for j2 in range(j1,m+1):
        cal  = s[i2][j2]-s[i2][j1-1]-s[i1-1][j2]+s[i1-1][j1-1]
        if cal<= k: ans+=1

print(ans)

2.子串分值

 把问题转换为单个字符能够创造的的价值求和,就是他当前x下标与前后x下标内下标的组合

考查:思维

s = ' ' + input()
n = len(s) - 1

index = {}

# 计算每一个元素能创造的价值
# 累计每一个元素的价值

for i in range(1, n + 1):
    c = s[i]

    if c not in index:
        index[c] = [0]
    
    index[c].append(i) #记录出现的位置


cnt = 0
for value in index.values():
    value.append(n + 1)
    # size = len(value)-2
    for i in range(1, len(value) - 1):
        cnt += (value[i] - value[i - 1]) * (value[i + 1] - value[i])    # 有多少取的种方法,就是它能创造的组合

print(cnt)

3.蓝桥杯国赛题目机器人塔

DFS搜索或者循环来做,可以学习Python缓存lru_cache的用法,即内置记忆化存储,用了这个可以加快运算效率,避免重复运算。

这道题结合二进制,同时从下到上递推,然后用了位运算等知识。

'''
# python3.6
# -*- coding: utf-8 -*-
# @Time    : 2023/5/12 13:01
# @Author  : Jin
# @File    : test2.py
# @Software: PyCharm
import os
import sys
import math

# 请在此输入您的代码
#机器人塔
from functools import lru_cache
a,b=(int(i) for i in input().split())
n=int(math.sqrt((a+b)*2))

@lru_cache(None)
def f(n,x,a,b):
    if n==1:  # 出去边界
        if x==0:
            return a-1==0
        else:
            return b-1==0
    if a<0 or b<0 :  # 直接剪枝
        return
    #print(bin(x))
    cnt=bin(x)[2:].count('1')
    a=a-(n-cnt)
    b=b-cnt
    next=(x^(x>>1))&((2**(n-1)-1))
    #print(bin(next))
    return f(n-1,next,a,b)
c=0
for m in range(0,1<<n):
    if f(n,m,a,b):  # 层数,状态,机器人数量a 0,机器人数量b 1
        c+=1
print(c)

'''
import os
import sys

# A-AA/BB; B-AB/BA,A-0; B-1;符合异或运算
m, n = map(int, input().split())

def check(now, num):
    num_a, num_b = 0, 0
    for i in range(num, 1, -1):  # i是层数也是机器人个数
        ls = list(bin(now))[2:]
        num_b += ls.count('1')
        num_a += i - ls.count('1')
        # 求上一层的状态
        # now右移一位相当于把now最低位的状态移走,再和原本的now异就得到上层结果,但是最高位要通过与运算去掉,
        # 这相当于原本的now除了最高位其他每一位都和前一位进行了异或,结果为1说明可以站B,为0站A
        now ^= (now >> 1)
        now &= (2**(i - 1) - 1)  # 移除now最高位的状态
    if now == 1: num_b += 1
    else: num_a += 1
    return num_a == m and num_b == n

# num*(num+1)//2 = m+n
num = int(((m + n) * 2) ** 0.5)  # 层数0~num-1 ,最后一层的个数
ans = 0
for i in range(1 << num):  # 最下面一层有num个数,有00..~11..种状态
    if check(i, num): ans += 1
print(ans)

4.跳蚂蚱

BFS模板题,主要是这个可以使用双向广搜,BFS双向广搜,了解一下写法。注意有时候BFS需要结合copy库来使用

import collections
import sys
meet=False   #判断是否相遇

def extend(q,m1,m2):
  global meet
  s=q.popleft()
  ss=list(s)
  index=ss.index('0') #找0的索引
  for j in [1,-1,2,-2]:
    ss[(index+j+9)%9],ss[index]= ss[index],ss[(index+j+9)%9]
    a=''.join(ss)
    if a in m2:  # 找到相同的了,即相遇了
      print(m1[s]+1+m2[a])
      meet=True
      return
    if a not in m1:
      q.append(a)
      m1[a]=m1[s]+1  # 向字典中添加
    ss[(index+j+9)%9],ss[index]= ss[index],ss[(index+j+9)%9]
 # meet=False

q1=collections.deque()
q2=collections.deque()
q1.append("012345678")
q2.append("087654321")
# 字典去重,同时记录步数
mp1={'012345678':0} ; mp2={'087654321':0}
while q1 and q2 :  # 两个都不为空
  if len(q1)<=len(q2): extend(q1,mp1,mp2)
  else:  extend(q2,mp2,mp1)
  if meet==True: break

5.树状数组处理逆序对

逆序对,两种方法,前面比他大的(正序树状数组),后面比他小的(倒叙树状数组),

import os
import sys

# 每个人的交换次数等前面严格大于自身的人数+后面严格小于自身的人数 数据范围是10^5暴力枚举O(n^2)会超时,此时就想到要用的树状数组或归并排序
#只有60%
# 请在此输入您的代码
N=1000010
def discretization(h):#数组离散化
    temp=list(set(h))
    temp.sort()
    for i in range(len(h)):
        h[i]=temp.index(h[i])+1
def lowbit(x):
    return x&-x
def update(x,d):
    while x<=N:
        tree[x]+=d
        x+=lowbit(x)
def sum(x):
    ans=0
    while x>0:
        ans+=tree[x]
        x-=lowbit(x)
    return ans

n=int(input())
hold=list(map(int,input().split()))  #身高
h=[0 for i in range(N)]  
for i in range(n):  # 转为下标从1开始
    h[i+1]=hold[i]
discretization(h) #数组离散化,但是因为本题的数字没超出范围就不需要了
k=[0 for i in range(N)]#每个小朋友要交换的次数
tree=[0 for i in range(N)] # 树状数组
for i in range(1,n+1): #正序处理,他后面的逆序对数量
    update(h[i],1)
    k[i]=i-sum(h[i])  # 计算h[i]前面的逆序数,前面有i-sum(h[i])个比他大的
   
tree=[0 for i in range(N)]
for i in range(n,0,-1): #倒序处理,他前面的逆序对数量
    k[i]+=sum(h[i]-1)  #计算h[i],后面有sum(h[i]-1)个比他小的
    update(h[i],1)
res=0  
for i in range(1,n+1):
    res+=int((1+k[i])*k[i]/2)  #等差数列
print(res)

归并排序处理逆序对

def merge_sort(L,R):
    if L < R:
        mid = (L+R)//2
        merge_sort(L,mid)
        merge_sort(mid+1,R)
        merge(L,mid,R)
def merge(L,mid,R):
    global res   # 记录答案
    i=L;j=mid+1;t=0
    while(i<=mid and j<=R):  #归并
        a[i]
        a[j]
        if (a[i]>a[j]):     #4 5 / 2 3   L=0 mid=1,R=3
            b[t]=a[j];t+=1;j+=1;
            res = res+(mid-i+1)  # 记录逆序对数量
        else:
            b[t] = a[i];t += 1;i += 1
    # 其中一个处理完了,把剩下的复制过来,直接整体复制
    # 这里注意区间取值,采用的是整体复制的思想,b是辅助数组
    if i<=mid: b[t:R-L+1]=a[i:mid+1]  # 取不到mid+1
    elif j<=R:b[t:R-L+1]=a[j:]
    # 把排序好的b[]复制回去a[]
    a[L:R+1]=b[:R-L+1]
n= int(input())
a = list(map(int,input().split()))
b = [0]*n
res = 0
merge_sort(0,n-1)
print(res)

6.数组切分

 DP的做法,python效率低所以不能AC

 

# 超时了两个,80%,思路没问题 
mod = 1000000007
N = 10010


f = [0 for i in range(N)]
n = int(input())
a =[0]+ list(map(int,input().split()))

f[0]=1
for i in range(1,n+1):
    mx=a[i]
    mi=a[i]
    for j in range(i,0,-1):
        mx = max(mx,a[j])
        mi = min(mi,a[j])
        if mx-mi == i-j:
            f[i] = (f[i]+f[j-1]) % mod   #dp思想  一个指针依次遍历i,另一个从当前位置回滚回去,长度从1变大,利用前一步结果计算
print(f[n])

猜你喜欢

转载自blog.csdn.net/weixin_52261094/article/details/130497182