gym100502 G - Outing （tarjan+分组背包）

题意

一辆车容量为V，现有n个人，每个人都有一个对应a[i]，表示只有a[i]上车 i才肯上车，问这辆车最多承载多少人。

思路 

由于a[i]和i的对应关系 可能存在树 或者 环 或者 环连着链->树 三种情况，我们可以得知如果存在环 这个环必然是所在树的树根。

因此对图求强连通分量，每个树所采纳的人数的【下限——树根的辣个强连通分量，上限——全部】，然后对这些树做分组背包求容量V以内的最大人数。

ACcode

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
 
using namespace std;
const int MaxN = 1e3 + 5;
typedef long long LL;
const LL mod = 1e9 + 7;
 
int w[MaxN],stack[MaxN],dfn[MaxN],low[MaxN],col[MaxN];
int n,m,cnt = 0,Top = 0,haha = 0,V;
bool vis[MaxN],ff[MaxN];
vector<int> G[MaxN],E[MaxN];
int v[MaxN],Size[MaxN],dp[MaxN],cur[MaxN],c[MaxN];
 
void tarjan(int u){
	dfn[u] = ++cnt;
	low[u] = dfn[u];
	stack[++Top] = u;
	vis[u] = 1;//在stack里
	int len = G[u].size();
	for(int i = 0;i < len; i++){
		int v = G[u][i];
		if(!dfn[v]){//没访问过
			tarjan(v);
			low[u] = min(low[u],low[v]);
		}
		else if(vis[v]){//访问过 && 在栈里
			low[u] = min(low[u],dfn[v]);
		}
	}
	if(low[u] == dfn[u]){
		haha++;
		int Si = 0;
		while(1){
			int cur = stack[Top];
			vis[cur] = 0;
			Top--;
			Si++;
			col[cur] = haha;
			if(cur == u) break;
		}
//		cout << Si << "?\n";
		Size[haha] = Si;
	}
}

void find(int x,int h){
	int N = G[x].size();
	for(int i = 0;i < N; i++){
		if(ff[G[x][i]]) continue;
		ff[G[x][i]] = 1;
		cur[h]++;
		find(G[x][i],h);
	}
}

int main()
{
	scanf("%d %d",&n,&V);
	for(int i = 1;i <= n; i++){
		int u;
		scanf("%d",&u);
		G[u].push_back(i);//i -> u
		//G[i的前置乘客].push_back(i);
	}
	for(int i = 1;i <= n; i++){
		if(!dfn[i]) tarjan(i);
	}//搜索所有强联通分量
//	for(int i = 1;i <= n; i++) cout << col[i] << " ";
//	cout << endl;
//	找出分组
	for(int i = 1;i <= n; i++){
		int len = G[i].size();
		for(int j = 0;j < len; j++){
			if(col[G[i][j]] == col[i]) continue;
			v[col[G[i][j]]]++;
		}
	}
	int K = 0;//k组
//	cout << haha << "###\n";
	for(int i = 1;i <= n; i++){
		if(!v[col[i]]){
			v[col[i]] = 1;
			K++;
			c[K] = Size[col[i]];
			cur[K] = 0;
			find(i,K);
		//	cout << c[K] << "#" << cur[K] << "\n";
		}
	}
//	for(int i = 1;i <= K; i++){
//		cout << i << ":" << c[i] << " " << cur[i] << endl;
//	}
	for(int k = 1;k <= K; k++){//k group
		for(int v = V;v >= 0; v--){
			for(int i = c[k];i <= cur[k]; i++){
				if(v >= i) dp[v] = max(dp[v],dp[v - i] + i);
			}
		}
	}
	printf("%d\n",dp[V]);
}