宝石收集,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)
相关推荐
亦皓ai17 分钟前
AI时代后端工程(二):AI把代码写得越来越快,我却越来越不敢让它直接开工了
人工智能·算法·机器学习·搜索引擎·transformer
玖玥拾2 小时前
LeetCode 392 判断子序列
笔记·算法·leetcode
重生之后端学习3 小时前
239. 滑动窗口最大值[困难]✅
java·数据结构·算法·leetcode·职场和发展
fpcc3 小时前
算法和数据结构—动态规划法
数据结构·算法·动态规划
星星.7226 小时前
C++算法竞赛|二分查找与二分答案:边界模板、浮点二分、STL
数据结构·c++·算法
sel_96 小时前
【OPD论文导读(二)】OPD(On-Policy Distillation)全景调研:十篇论文讲透“在自己生成的内容上学习“这件事
人工智能·python·深度学习·学习·算法·语言模型
金幄科技7 小时前
RAG第四篇:重排序 Rerank——粗召回之后,为什么还需要一次精排
人工智能·算法·ai·ai编程
M78佐菲7 小时前
Linux学习笔记:文件IO
linux·笔记·学习·算法
lhldsg8 小时前
课程排课系统实战指南:从数据库设计到算法调优全流程解析
java·数据库·算法·小程序
BSD_HY8 小时前
薄膜开关矩阵扫描电路设计中的防鬼键措施
人工智能·算法·矩阵·人机交互·薄膜开关·源头工厂·深圳工厂