【HDU 4547】 CD操作

【題目鏈接】

          點擊打開鏈接

【算法】

          分四種情況討論 :

          1. 當前目錄和目標目錄是同一目錄,不需要變換,答案爲0

          2. 當前目錄是目標目錄的祖先,答案爲當前目錄的深度 - 目標目錄的深度

          3. 當前目錄是目標目錄的孩子,答案爲1

          4. 當前目錄和目標目錄有最近公共祖先,答案爲當前目錄的深度 - 最近公共祖先的深度 + 1

          那麼,算法就很明確了,先做一遍離線tarjan,求出詢問點對的最近公共祖先,然後,進行上述的分類討論

【代碼】

          由於目錄名稱是字符串,筆者用stl庫裏的map將這些目錄重新編號

          

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100010
#define MAXL 45

struct Edge
{
		int to,nxt;
} e[MAXN];

int i,T,tot,id,n,m,root;
int head[MAXN],dep[MAXN],f[MAXN],lca[MAXN],fa[MAXN];
string a,b;
string x[MAXN],y[MAXN];
bool visited[MAXN];
map<string,int> mp;
vector< pair<int,int> > query[MAXN];

inline void add(int u,int v)
{
		tot++;
		e[tot] = (Edge){v,head[u]};
		head[u] = tot;
}
inline void init(int u)
{
		int i,v;
		for (i = head[u]; i; i = e[i].nxt)
		{
				v = e[i].to;
				dep[v] = dep[u] + 1;
				init(v);
		}		
}
inline int find(int x)
{
		if (f[x] == x) return x;
		return f[x] = find(f[x]);
}
inline void tarjan(int u)
{
		int i,v,pos;
		visited[u] = true;
		f[u] = u;
		for (i = 0; i < query[u].size(); i++)
		{
			  v = query[u][i].first;
			  pos = query[u][i].second;
			  if (visited[v]) lca[pos] = find(v);
		}
		for (i = head[u]; i; i = e[i].nxt)
		{
				v = e[i].to;
				tarjan(v);
				f[v] = u;
		}
}

int main() 
{
		
		ios :: sync_with_stdio(0);
		cin.tie(0);
		
		cin >> T;
		while (T--)
		{
				cin >> n >> m;
				tot = id = 0;
				mp.clear();
				for (i = 1; i <= n; i++)
				{
						head[i] = 0;
						fa[i] = 0;
						visited[i] = false;
						query[i].clear();
				}
				for (i = 1; i < n; i++) 
				{
						cin >> a >> b;
						if (!mp[a]) mp[a] = ++id;
						if (!mp[b]) mp[b] = ++id;	
						add(mp[b],mp[a]);
						fa[mp[a]] = mp[b];
				}	
				for (i = 1; i <= n; i++) 
				{
						if (!fa[i])
								root = i;
				}
				dep[root] = 0;
				init(root);
				for (i = 1; i <= m; i++) 
				{
						cin >> x[i] >> y[i];
						query[mp[x[i]]].push_back(make_pair(mp[y[i]],i));
						query[mp[y[i]]].push_back(make_pair(mp[x[i]],i));
				}
				tarjan(root);
				for (i = 1; i <= m; i++) 
				{
						if (x[i] == y[i]) printf("%d\n",0);
						else if (lca[i] == mp[x[i]]) printf("%d\n",1);
						else if (lca[i] == mp[y[i]]) printf("%d\n",dep[mp[x[i]]]-dep[mp[y[i]]]);
						else printf("%d\n",dep[mp[x[i]]]-dep[lca[i]]+1);
				}
		}
		
		return 0;
	
}

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