题目描述:
题解:
比赛的时候看出来跟强连通分量有关了。庆幸自己硬着头皮写了强连通分量了,图论确实练的不多。
我们两两对应建图。可以发现有两种情况:
1.当强连通分量里面元素个数为2的时候,我们可以找这个这两个元素的反向最长链。因为有两个人相互喜欢的话,它可以无限延伸出去。我们dfs找两条最长链相加就行。同时所有答案可以相加,因为已经成为闭环了。
2.当强连通大于2时,桌子就只能坐下这个强连通分量的所有人了。
并且最多只能坐一个了。
AC代码:
class Solution {
public:
vector<int>G[100005],P[100005],scc[100005];
int dfn[100005],low[100005],vis[100005],cnt,siz[100005],tot;
stack<int>st;
void tarjan(int u)
{
st.push(u);
dfn[u]=low[u]=++cnt;
vis[u]=1;
for(auto v:G[u])
{
if(!dfn[v])
{
tarjan(v);
low[u]=min(low[u],low[v]);
}
else if(vis[v])
{
low[u]=min(low[u],dfn[v]);
}
}
if(dfn[u]==low[u])
{
tot++;
int now;
do{
now=st.top();st.pop();
scc[tot].push_back(now);
siz[tot]++;
vis[now]=0;
}while(now!=u);
}
}
int dfs(int u,int x)
{
int num=0;
for(auto v:P[u])
{
if(v!=x)num=max(dfs(v,x),num);
}
return num+1;
}
int maximumInvitations(vector<int>& favorite) {
int len=favorite.size();
for(int i=0;i<len;i++)
{
G[i].push_back(favorite[i]);
P[favorite[i]].push_back(i);
}
for(int i=0;i<len;i++)
{
if(!dfn[i])tarjan(i);
}
int ans=0,res=0,f=0;
for(int i=1;i<=tot;i++)
{
if(siz[i]==2)
{
int num=dfs(scc[i][0],scc[i][1])+dfs(scc[i][1],scc[i][0]);
ans+=num;
}
}
for(int i=1;i<=tot;i++)
{
if(siz[i]>2) ans=max(ans,siz[i]);
}
return ans;
}
};