... | ... |
@@ -88,13 +88,13 @@ |
88 | 88 |
ti.restart(); |
89 | 89 |
pre.run(); |
90 | 90 |
if(report) std::cerr << "Run Preflow: " << ti << '\n'; |
91 | 91 |
if(report) std::cerr << "\nMax flow value: " << pre.flowValue() << '\n'; |
92 | 92 |
} |
93 | 93 |
|
94 |
template<class Value> |
|
94 |
template<class Value, class LargeValue> |
|
95 | 95 |
void solve_min(ArgParser &ap, std::istream &is, std::ostream &, |
96 | 96 |
Value infty, DimacsDescriptor &desc) |
97 | 97 |
{ |
98 | 98 |
bool report = !ap.given("q"); |
99 | 99 |
Digraph g; |
100 | 100 |
Digraph::ArcMap<Value> lower(g), cap(g), cost(g); |
... | ... |
@@ -124,13 +124,14 @@ |
124 | 124 |
if (report) std::cerr << "Setup NetworkSimplex class: " << ti << '\n'; |
125 | 125 |
ti.restart(); |
126 | 126 |
bool res = ns.run(); |
127 | 127 |
if (report) { |
128 | 128 |
std::cerr << "Run NetworkSimplex: " << ti << "\n\n"; |
129 | 129 |
std::cerr << "Feasible flow: " << (res ? "found" : "not found") << '\n'; |
130 |
if (res) std::cerr << "Min flow cost: " |
|
130 |
if (res) std::cerr << "Min flow cost: " |
|
131 |
<< ns.template totalCost<LargeValue>() << '\n'; |
|
131 | 132 |
} |
132 | 133 |
} |
133 | 134 |
|
134 | 135 |
void solve_mat(ArgParser &ap, std::istream &is, std::ostream &, |
135 | 136 |
DimacsDescriptor &desc) |
136 | 137 |
{ |
... | ... |
@@ -148,13 +149,13 @@ |
148 | 149 |
if(report) std::cerr << "Run MaxMatching: " << ti << '\n'; |
149 | 150 |
if(report) std::cerr << "\nCardinality of max matching: " |
150 | 151 |
<< mat.matchingSize() << '\n'; |
151 | 152 |
} |
152 | 153 |
|
153 | 154 |
|
154 |
template<class Value> |
|
155 |
template<class Value, class LargeValue> |
|
155 | 156 |
void solve(ArgParser &ap, std::istream &is, std::ostream &os, |
156 | 157 |
DimacsDescriptor &desc) |
157 | 158 |
{ |
158 | 159 |
std::stringstream iss(static_cast<std::string>(ap["infcap"])); |
159 | 160 |
Value infty; |
160 | 161 |
iss >> infty; |
... | ... |
@@ -166,13 +167,13 @@ |
166 | 167 |
exit(1); |
167 | 168 |
} |
168 | 169 |
|
169 | 170 |
switch(desc.type) |
170 | 171 |
{ |
171 | 172 |
case DimacsDescriptor::MIN: |
172 |
solve_min<Value>(ap,is,os,infty,desc); |
|
173 |
solve_min<Value, LargeValue>(ap,is,os,infty,desc); |
|
173 | 174 |
break; |
174 | 175 |
case DimacsDescriptor::MAX: |
175 | 176 |
solve_max<Value>(ap,is,os,infty,desc); |
176 | 177 |
break; |
177 | 178 |
case DimacsDescriptor::SP: |
178 | 179 |
solve_sp<Value>(ap,is,os,desc); |
... | ... |
@@ -261,17 +262,19 @@ |
261 | 262 |
std::cout << "\nNum of nodes: " << desc.nodeNum; |
262 | 263 |
std::cout << "\nNum of arcs: " << desc.edgeNum; |
263 | 264 |
std::cout << "\n\n"; |
264 | 265 |
} |
265 | 266 |
|
266 | 267 |
if(ap.given("double")) |
267 |
solve<double>(ap,is,os,desc); |
|
268 |
solve<double, double>(ap,is,os,desc); |
|
268 | 269 |
else if(ap.given("ldouble")) |
269 |
solve<long double>(ap,is,os,desc); |
|
270 |
solve<long double, long double>(ap,is,os,desc); |
|
270 | 271 |
#ifdef LEMON_HAVE_LONG_LONG |
271 | 272 |
else if(ap.given("long")) |
272 |
solve<long long>(ap,is,os,desc); |
|
273 |
solve<long long, long long>(ap,is,os,desc); |
|
274 |
else solve<int, long long>(ap,is,os,desc); |
|
275 |
#else |
|
276 |
else solve<int, long>(ap,is,os,desc); |
|
273 | 277 |
#endif |
274 |
else solve<int>(ap,is,os,desc); |
|
275 | 278 |
|
276 | 279 |
return 0; |
277 | 280 |
} |
0 comments (0 inline)