X-Git-Url: https://ruin.nu/git/?a=blobdiff_plain;f=src%2Fgenealgorithms.cpp;h=7f54109dc6c0103251cafff77bef93d92fa9f1e4;hb=148a31960aeb2ba127ef4322a12ae3af28ef2988;hp=eecdf3fea92bd8048c6c47fd1ef466d5e4698c67;hpb=6be4116d6211ac4ba4c68dc2ab28f1bc5d77068e;p=germs.git

diff --git a/src/genealgorithms.cpp b/src/genealgorithms.cpp
index eecdf3f..7f54109 100644
--- a/src/genealgorithms.cpp
+++ b/src/genealgorithms.cpp
@@ -23,6 +23,7 @@
 
 #include <algorithm>
 #include <set>
+#include <stack>
 #include <cstdlib>
 #include <iostream>
 using namespace std;
@@ -56,47 +57,150 @@ std::vector<std::vector<int> > robinsonSchensted(const GeneOrder& go){
 	return v;
 }
 
-int countCycles(const GeneOrder& go){
-	int cycles = 0;
+struct FindP{
+	size_t p;
+	FindP(size_t p) : p(p) {}
+	bool operator()(Interval i){
+		return (i.first == p || i.second == p);
+	}
+};
+
+
+size_t countCycles(const GeneOrder& go){
+	size_t cycles = 0;
 	set<size_t> marked;
-	for (size_t p = 1; p < go.size() - 1; ++p){
+	vector<Interval> intervals = findIntervals(go);
+	vector<Interval> points = findIntervalsAtPoints(intervals);
+	for (size_t p = 1; p < go.size(); ++p){
 		if (marked.find(p) != marked.end())
 			continue;
+		Interval i = intervals[points[p].first];
+		while (marked.find(p) == marked.end()){
+			marked.insert(p);
+			if (i == intervals[points[p].first])
+				i = intervals[points[p].second];
+			else
+				i = intervals[points[p].first];
+
+			if (p == i.first)
+				p = i.second;
+			else
+				p = i.first;
+		}
+		++cycles;
 	}
 	return cycles;
 }
 
+int sign(Gene g){
+	if (g > 0)
+		return 1;
+	if (g < 0)
+		return -1;
+	return 0;
+}
+
+struct Abs{
+	Gene operator()(Gene x) const{
+		return abs(x);
+	}
+};
 std::vector<Component> findComponents(const GeneOrder& go){
-	return vector<Component>();
+	vector<Component> components;
+	vector<int> os(go.size()-1);
+	for (size_t i = 0; i < os.size(); ++i)
+		os[i] = (go[i]*go[i+1] > 0 ? sign(go[i]) : 0);
+	stack<Gene> Mdir;
+	Mdir.push(go.size()-1);
+	stack<Gene> Mrev;
+	Mrev.push(0);
+	stack<size_t> Sdir;
+	Sdir.push(0);
+	stack<size_t> Srev;
+	Srev.push(0);
+	vector<Gene> dir;
+	dir.push_back(go.size()-1);
+	vector<Gene> rev;
+	rev.push_back(0);
+	size_t s;
+	vector<Gene> p(go.list());
+	transform(p.begin(),p.end(),p.begin(),Abs());
+	for (size_t i = 1; i < go.size(); ++i){
+		//Directed
+		if (p[i-1] > p[i])
+			Mdir.push(p[i-1]);
+		else while (Mdir.top() < p[i])
+			Mdir.pop();
+		dir.push_back(Mdir.top());
+
+		s = Sdir.top();
+		while(p[Sdir.top()] > p[i] || dir[Sdir.top()] < p[i]){
+			Sdir.pop();
+			os[Sdir.top()] = (os[Sdir.top()] == os[s] ? os[s] : 0);
+			s = Sdir.top();
+		}
+		if (go[i] > 0 && dir[i] == dir[s] && static_cast<Gene>(i - s) == p[i] - p[s])
+			components.push_back(Component(p[s],p[i],(s+1 == i ? 0 : os[s])));
+
+		//Reverse
+		if (p[i-1] < p[i])
+			Mrev.push(p[i-1]);
+		else while (Mrev.top() > p[i])
+			Mrev.pop();
+		rev.push_back(Mrev.top());
+
+		s = Srev.top();
+		while((p[s] < p[i] || rev[s] > p[i]) && s > 0){
+			Srev.pop();
+			os[Srev.top()] *= (os[Srev.top()] == os[s] ? 1 : 0);
+			s = Srev.top();
+		}
+		if (go[i] < 0 && rev[i] == rev[s] && static_cast<Gene>(i - s) == p[s] - p[i])
+			components.push_back(Component(-p[s],-p[i],(s+1 == i ? 0 : os[s])));
+
+		//Update stacks
+		if (go[i] > 0)
+			Sdir.push(i);
+		else
+			Srev.push(i);
+	}
+	return components;
 }
 
-//TODO: Think of a better than O(n^2) implementation
+/**
+ * 
+ */
 std::vector<Interval> findIntervals(const GeneOrder& go){
-	vector<Interval> intervals;
-	for (size_t i = 0; i < go.size() - 1; ++i){
-		size_t f = 0;
-		size_t s = 0;
-		bool found = false;
-		size_t n = 0;
-		for (GeneOrder::iterator g = go.begin(); g != go.end(); ++g, ++n){
-			if (static_cast<size_t>(abs(*g)) == i){
-				f = n;
-				if (*g >= 0)
-					++f;
-				if (found)
-					break;
-				found = true;
-			}
-			if(static_cast<size_t>(abs(*g)) == i+1){
-				s = n;
-				if (*g < 0)
-					++s;
-				if (found)
-					break;
-				found = true;
-			}
-		}
-		intervals.push_back(Interval(f,s));
+	vector<Interval> intervals(go.size()-1,Interval(go.size(),go.size()));
+	size_t n = 0;
+	for (GeneOrder::iterator g = go.begin(); g != go.end(); ++g, ++n){
+			size_t i = abs(*g);
+			if (i < go.size() - 1)
+				intervals[i].first = n + (*g >= 0 ? 1 : 0);
+			if (i > 0)
+				intervals[i-1].second = n + (*g < 0 ? 1 : 0);
 	}
 	return intervals;
 }
+
+/**
+ * 
+ */
+std::vector<Interval> findIntervalsAtPoints(const vector<Interval>& intervals){
+	size_t max = intervals.size()+1;
+	vector<Interval> points(max,Interval(max,max));
+	size_t n = 0;
+	for (vector<Interval>::const_iterator i = intervals.begin(); i != intervals.end(); ++i, ++n){
+		if (points[i->first].first == max){
+			points[i->first].first = n;
+		}else
+			points[i->first].second = n;
+			
+		if (points[i->second].first == max){
+			points[i->second].first = n;
+		}else
+			points[i->second].second = n;
+	}
+	return points;
+
+}