首页 > 代码库 > poj 1470 Closest Common Ancestors 【Tarjan 离线 LCA】
poj 1470 Closest Common Ancestors 【Tarjan 离线 LCA】
题目:poj 1470 Closest Common Ancestors
题意:给出一个树,一些询问。求LCA的个数、
分析:很简单的模板题目,但是模板不够优秀,一直wa...RE,各种错误一下午,终于发现自己模板的漏洞了。
AC代码:
#include <iostream> #include <cstdio> #include <cstring> #include <vector> using namespace std; #define N 1010 #define M 20010 struct Node { int to,val; }; vector<Node> g[N]; struct ask{ int u,v,lca; }; vector<ask> e; vector<int> query[N]; int fa[N],ance[N],dir[N]; bool vis[N]; inline void add_Node(int u ,int v,int w) { g[u].push_back((Node){v,w}); g[v].push_back((Node){u,w}); } inline void add_ask(int u ,int v ) { e.push_back((ask){u,v,-1}); e.push_back((ask){v,u,-1}); int len = e.size()-1; query[v].push_back(len); query[u].push_back(len-1); } int find(int x){ return x == fa[x] ? x : fa[x] = find(fa[x]); } void Tarjan(int u,int val) { vis[u] = true; ance[u] = fa[u] = u; dir[u] = val; for(int i=0;i<g[u].size();i++) { Node tmp = g[u][i]; if(!vis[tmp.to]) { Tarjan(tmp.to,val+tmp.val); fa[tmp.to] = u; } } for(int i=0;i<query[u].size();i++) { int num = query[u][i]; ask& tmp = e[num]; if(vis[tmp.v]) { tmp.lca = e[num^1].lca = ance[find(tmp.v)]; } } } int ans[N],flag[N]; void Clear(int n) { memset(vis,false,sizeof(vis)); memset(ans,0,sizeof(ans)); memset(flag,0,sizeof(flag)); for(int i=0;i<=n;i++){ g[i].clear(); query[i].clear(); } e.clear(); } int main() { //freopen("Input.txt","r",stdin); int n; int u,x,m; char ch; while(scanf("%d",&n)!=EOF) { Clear(n); for(int i=1;i<=n;i++) { scanf("%d:(%d)",&u,&m); while(m--) { scanf("%d",&x); flag[x]=true; add_Node(x,u,1); } } int Q; scanf("%d",&Q); for(int i=1;i<=Q;i++) { cin>>ch; scanf("%d%d",&u,&x); cin>>ch; add_ask(u,x); } int root; for(int i=1;i<=n;i++) if(!flag[i]) { root=i; break; } Tarjan(root,0); for(int i = 0;i< Q;i++) { ans[e[i*2].lca]++; } for(int i = 1; i<= N;i++) { if(ans[i]) printf("%d:%d\n",i,ans[i]); } } return 0; }
poj 1470 Closest Common Ancestors 【Tarjan 离线 LCA】
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。