宝石收集,tarjan

0宝石收集 - 蓝桥云课 (lanqiao.cn)

复制代码
n=int(input())
s='0'+input()
m=int(input())
mp=[[] for i in range(n+1)]
for i in range(m):
    a,b=map(int,input().split())
    a+=1
    b+=1

    mp[a].append(b)


import sys
sys.setrecursionlimit(100000000)
dfn=[0 for i in range(n+1)]
low=[0 for i in range(n+1)]
cnt=0

stk=[]
instk=[0 for i in range(n+1)]

p=0
scc=[0 for i in range(n+1)]

def tarjan(x):
    global cnt,p
    cnt+=1
    dfn[x]=low[x]=cnt
    stk.append(x)
    instk[x]=1
    for i in mp[x]:
        if dfn[i]==0:
            tarjan(i)
            low[x]=min(low[x],low[i])
        elif instk[i]:
            low[x]=min(low[x],dfn[i])


    if dfn[x]==low[x]:
        p+=1
        y=stk.pop()
        instk[y]=0
        scc[y]=p
        while x!=y:
            y = stk.pop()
            instk[y] = 0
            scc[y] = p
for i in range(1,n+1):
    if dfn[i]==0:tarjan(i)

del instk,low,dfn

ru=[0 for i in range(p+1)]
h=[0 for i in range(p+1)]
l=[0 for i in range(p+1)]
nmp=[[] for i in range(p+1)]
for i in range(1,n+1):
    if s[i]=='0': h[scc[i]]+=1
    else : l[scc[i]]+=1
    for j in mp[i]:
        if scc[i]!=scc[j]:
            ru[scc[j]]+=1
            nmp[scc[i]].append(scc[j])
del mp

def dfs(x,a,b):
    ans=0
    vis=0
    for y in nmp[x]:
        ans=max(dfs(y,a+h[y],b+l[y]),ans)
        vis=1
    if vis==0:return min(a,b)
    else:return ans
ans=0
for i in range(1,p+1):
    if ru[i]==0:
        ans=max(ans,dfs(i,h[i],l[i]))
print(ans)
相关推荐
修炼地2 分钟前
代码随想录算法训练营第四十三天 | 图论理论基础、深搜理论基础、卡码网98. 所有可达路径、797. 所有可能的路径、广搜理论基础
算法·深度优先·图论
iAkuya2 分钟前
(leetcode)力扣100 23反转链表(迭代||递归)
算法·leetcode·链表
剪一朵云爱着4 分钟前
PAT 1095 Cars on Campus
算法·pat考试
MicroTech20251 小时前
激光点云快速配准算法创新突破,MLGO微算法科技发布革命性点云配准算法技术
人工智能·科技·算法
Cathy Bryant1 小时前
傅里叶变换(一):简介
笔记·算法·数学建模·信息与通信·傅里叶分析
allan bull2 小时前
在节日中寻找平衡:圣诞的欢乐与传统节日的温情
人工智能·学习·算法·职场和发展·生活·求职招聘·节日
似水এ᭄往昔2 小时前
【C++】--封装红⿊树实现mymap和myset
开发语言·数据结构·c++·算法·stl
咕噜企业分发小米2 小时前
腾讯云向量数据库HNSW索引如何更新?
人工智能·算法·腾讯云
lcreek2 小时前
LeetCode215. 数组中的第K个最大元素、LeetCode912. 排序数组
python·算法·leetcode
Einsail3 小时前
天梯赛题解(3-6)
算法