Codeforces 427C - Checkposts 極大連通分量Tarjan算法模板題(STL實現)

               這是一道,求極大連通分量的模板題,樓主第一次寫,網上搜到的模板全部都是用數組存的,竟然沒有找到用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;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章