寻找共同的父节点

git是一种分布式代码管理工具，git通过树的形式记录文件的更改历史，比如： base'<--base<--A<--A' ^ | --- B<--B' 小米工程师常常需要寻找两个分支最近的分割点，即base.假设git 树是多叉树，请实现一个算法，计算git树上任意两点的最近分割点。 （假设git树节点数为n,用邻接矩阵的形式表示git树：字符串数组matrix包含n个字符串，每个字符串由字符'0'或'1'组成，长度为n。matrix[i][j]=='1'当且仅当git树种第i个和第j个节点有连接。节点0为git树的根节点。）

参考了http://blog.csdn.net/shenxuan520/article/details/73832618

解题思路

从矩阵构造出父节点记录。
记录从根节点到带求节点A和B的路径。
比较路径，找到最近的公共节点。

#include <iostream>
#include <string>
#include <vector>
#include <stack>
using namespace std;

int getSplitNode(vector<string> matrix, int indexA, int indexB) {
stack<int> s;
if(indexA==indexB)
return indexA;
int len=matrix.size();
vector<int> father(len,-1);//父节点记录表
vector<int> flag(len,0);// 保证不重复入栈
s.push(0);//先将根节点入栈
flag[0] = 1;

int count = 0;
int temp;//记录当前栈顶
while(!s.empty()){
temp = s.top();
s.pop();
for(int count = 1; count<len;count++){
if(matrix[temp][count] == '1' && flag[count] ==0){
s.push(count);
flag[count] = 1;
father[count] = temp;//记录父节点
}
}
}

vector<int> patha;patha.push_back(indexA);
vector<int> pathb;pathb.push_back(indexB);

int ta = indexA;
int tb = indexB;

while(father[ta] != -1){
patha.push_back(father[ta]);
ta = father[ta];
}

while(father[tb] != -1){
pathb.push_back(father[tb]);
tb = father[tb];
}

ta = patha.size()-1;
tb = pathb.size()-1;
while(ta>= 0 && tb >= 0 && patha[ta] == pathb[tb]){
ta--;
tb--;
}

return pathb[tb+1];// or patha[ta+1]
}

int main() {
// your code goes here
vector<string> matrix;
matrix.push_back("0001");
matrix.push_back("0001");
matrix.push_back("0001");
matrix.push_back("1110");

cout<<getSplitNode(matrix,3,2)<<endl;
return 0;
}