bibd.cpp
Go to the documentation of this file.
00001 /* -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */ 00002 /* 00003 * Main authors: 00004 * Christian Schulte <schulte@gecode.org> 00005 * 00006 * Copyright: 00007 * Christian Schulte, 2004 00008 * 00009 * Bugfixes provided by: 00010 * Olof Sivertsson <olsi0767@student.uu.se> 00011 * 00012 * Last modified: 00013 * $Date: 2010-10-07 11:52:01 +0200 (Thu, 07 Oct 2010) $ by $Author: schulte $ 00014 * $Revision: 11473 $ 00015 * 00016 * This file is part of Gecode, the generic constraint 00017 * development environment: 00018 * http://www.gecode.org 00019 * 00020 * Permission is hereby granted, free of charge, to any person obtaining 00021 * a copy of this software and associated documentation files (the 00022 * "Software"), to deal in the Software without restriction, including 00023 * without limitation the rights to use, copy, modify, merge, publish, 00024 * distribute, sublicense, and/or sell copies of the Software, and to 00025 * permit persons to whom the Software is furnished to do so, subject to 00026 * the following conditions: 00027 * 00028 * The above copyright notice and this permission notice shall be 00029 * included in all copies or substantial portions of the Software. 00030 * 00031 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 00032 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 00033 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 00034 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE 00035 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 00036 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 00037 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 00038 * 00039 */ 00040 00041 #include <gecode/driver.hh> 00042 #include <gecode/int.hh> 00043 00044 using namespace Gecode; 00045 00050 class BIBDOptions : public Options { 00051 public: 00052 int v, k, lambda; 00053 int b, r; 00054 00055 void derive(void) { 00056 b = (v*(v-1)*lambda)/(k*(k-1)); 00057 r = (lambda*(v-1)) / (k-1); 00058 } 00060 BIBDOptions(const char* s, 00061 int v0, int k0, int lambda0) 00062 : Options(s), v(v0), k(k0), lambda(lambda0) { 00063 derive(); 00064 } 00066 void parse(int& argc, char* argv[]) { 00067 Options::parse(argc,argv); 00068 if (argc < 4) 00069 return; 00070 v = atoi(argv[1]); 00071 k = atoi(argv[2]); 00072 lambda = atoi(argv[3]); 00073 derive(); 00074 } 00076 virtual void help(void) { 00077 Options::help(); 00078 std::cerr << "\t(unsigned int) default: " << v << std::endl 00079 << "\t\tparameter v" << std::endl 00080 << "\t(unsigned int) default: " << k << std::endl 00081 << "\t\tparameter k" << std::endl 00082 << "\t(unsigned int) default: " << lambda << std::endl 00083 << "\t\tparameter lambda" << std::endl; 00084 } 00085 }; 00086 00087 00096 class BIBD : public Script { 00097 protected: 00099 const BIBDOptions& opt; 00101 BoolVarArray _p; 00102 public: 00104 BIBD(const BIBDOptions& o) 00105 : opt(o), _p(*this,opt.v*opt.b,0,1) { 00106 Matrix<BoolVarArray> p(_p,opt.b,opt.v); 00107 00108 // r ones per row 00109 for (int i=0; i<opt.v; i++) 00110 linear(*this, p.row(i), IRT_EQ, opt.r); 00111 00112 // k ones per column 00113 for (int j=0; j<opt.b; j++) 00114 linear(*this, p.col(j), IRT_EQ, opt.k); 00115 00116 // Exactly lambda ones in scalar product between two different rows 00117 for (int i1=0; i1<opt.v; i1++) 00118 for (int i2=i1+1; i2<opt.v; i2++) { 00119 BoolVarArgs row(opt.b); 00120 for (int j=0; j<opt.b; j++) 00121 row[j] = expr(*this, p(j,i1) && p(j,i2)); 00122 linear(*this, row, IRT_EQ, opt.lambda); 00123 } 00124 00125 // Symmetry breaking 00126 for (int i=1; i<opt.v; i++) 00127 rel(*this, p.row(i-1), IRT_GQ, p.row(i)); 00128 for (int j=1; j<opt.b; j++) 00129 rel(*this, p.col(j-1), IRT_GQ, p.col(j)); 00130 00131 branch(*this, _p, INT_VAR_NONE, INT_VAL_MIN); 00132 } 00133 00135 virtual void 00136 print(std::ostream& os) const { 00137 os << "\tBIBD(" 00138 << opt.v << "," << opt.k << "," 00139 << opt.lambda << ")" << std::endl; 00140 Matrix<BoolVarArray> p(_p,opt.b,opt.v); 00141 for (int i = 0; i<opt.v; i++) { 00142 os << "\t\t"; 00143 for (int j = 0; j<opt.b; j++) 00144 os << p(j,i) << " "; 00145 os << std::endl; 00146 } 00147 os << std::endl; 00148 } 00149 00151 BIBD(bool share, BIBD& s) 00152 : Script(share,s), opt(s.opt) { 00153 _p.update(*this,share,s._p); 00154 } 00155 00157 virtual Space* 00158 copy(bool share) { 00159 return new BIBD(share,*this); 00160 } 00161 00162 }; 00163 00167 int 00168 main(int argc, char* argv[]) { 00169 BIBDOptions opt("BIBD",7,3,60); 00170 opt.parse(argc,argv); 00171 00172 /* 00173 * Other interesting instances: 00174 * BIBD(7,3,1), BIBD(6,3,2), BIBD(7,3,20), ... 00175 */ 00176 00177 Script::run<BIBD,DFS,BIBDOptions>(opt); 00178 return 0; 00179 } 00180 00181 // STATISTICS: example-any 00182