Codeforces 427C - Checkposts 极大连通分量Tarjan算法模板题(STL实现)

来源:互联网 发布:python生成一组随机数 编辑:程序博客网 时间:2024/05/17 04:46

               这是一道,求极大连通分量的模板题,楼主第一次写,网上搜到的模板全部都是用数组存的,竟然没有找到用STL写的模板,于是楼主自己照着板子打了一遍。本题的思路其实很简单,找整个图里面的极大强连通分量,然后找到每个分量里面的最小代价,如果有多个就用数组存起来。当然了,要是不会求强连通分量的算法的话这道题自己构造一个dfs的算法就比较复杂了,基础还是要巩固滴<( ̄ˇ ̄)/,以下是AC代码:

#include<bits/stdc++.h>//强连通分量Tarjan算法模板题目 using namespace std;typedef long long ll;typedef struct Node{int w;vector<int>e;}Node;Node node[100010];int mod = 1e9+7; int index = 0;//记录访问的节点的顺序序号 int num = 0; //记录现在出现的极大强连通分量的序号 int dfn[100010];//记录是否访问的数组 int low[100010];//记录某一个节点在dfs的时间戳 int instack[100010];//记录一个节点是否在栈中int belong[100010];//记录某一个节点所在的强连通分量的序号int cost[100010];//记录每一个极大强连通分量的最小代价 int cnt[100010];//记录每一个极大强连通分量最小代价的可选择数 stack<int>sta;//栈 void init(){memset(dfn, 0, sizeof(dfn));memset(low, 0, sizeof(low));memset(instack, 0, sizeof(instack));memset(belong, 0, sizeof(belong));memset(cnt, 0, sizeof(cnt));memset(cost, 0x3f3f3f3f, sizeof(cost));}void Tarjan(int u){dfn[u] = low[u] = ++index;instack[u] = 1;//虽然已经进栈但还是要标记一下 sta.push(u);for(int i = 0; i < node[u].e.size(); i++){int t = node[u].e[i];if(!dfn[t]){Tarjan(t);low[u] = min(low[u], low[t]);}    else if(instack[t])    {    low[u] = min(low[u], dfn[t]);    }}if(dfn[u] == low[u]) //如果节点u是强连通分量的根{num++; while(1) //一个个出栈,这里不要写错啦 {int v = sta.top();instack[v] = 0;belong[v] = num;//printf("%d %d\n", v, belong[v]);sta.pop();if(v == u)   break;}} }int main(){int n, m;init(); scanf("%d", &n);for(int i = 1; i <= n; i++)   scanf("%d", &node[i].w);scanf("%d", &m);for(int i = 1; i <= m; i++){int a, b;scanf("%d%d", &a, &b);node[a].e.push_back(b);} for(int i = 1; i <= n; i++){if(!dfn[i])//如果没有访问过该节点则进行搜索    Tarjan(i);}for(int i = 1; i <= n; i++){//printf("%d\n", belong[i]);if(cost[belong[i]] > node[i].w){cost[belong[i]] = node[i].w;cnt[belong[i]] = 1;}else if(cost[belong[i]] == node[i].w){cnt[belong[i]]++;}} ll ans1 = 0, ans2 = 1; for(int i = 1; i <= num; i++){ans1 += (ll)cost[i];ans2 = (ans2*(ll)cnt[i])%mod;}printf("%I64d %I64d\n", ans1, ans2);return 0;}


1 0
原创粉丝点击