-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathdlx1.w
851 lines (764 loc) · 32.6 KB
/
dlx1.w
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
@s item int
@s node int
@s mod and
\let\Xmod=\bmod % this is CWEB magic for using "mod" instead of "%"
\datethis
@*Intro. This program is part of a series of ``exact cover solvers'' that
I'm putting together for my own education as I prepare to write Section
7.2.2.1 of {\sl The Art of Computer Programming}. My intent is to
have a variety of compatible programs on which I can run experiments,
in order to learn how different approaches work in practice.
Indeed, this is the first of the series.
I've tried to write it as a primitive baseline against which I'll be able
to measure various technical improvements and extensions.
{\mc DLX1} is based on the program {\mc DANCE}, which I wrote hastily
in 1999 while preparing my paper about ``Dancing Links.'' [See
{\sl Selected Papers on Fun and Games\/} (2011), Chapter~38, for a
revised version of that paper, which first appeared in the book
{\sl Millennial Perspectives in Computer Science\/}, a festschrift
for C.~A.~R. Hoare (2000).] That program, incidentally, was based
on a program called {\mc XCOVER} that I first wrote in 1994.
After using {\mc DANCE} as a workhorse
for more than 15 years, and after extending it in dozens of ways
for a wide variety of combinatorial problems, I'm finally ready to replace it
with a more carefully crafted piece of code.
My intention is to make this program match Algorithm 7.2.2.1X, so that
I~can use it to make the quantitative experiments that will ultimately
be reported in Volume~4B.
Although this is the entry-level program, I'm taking care to adopt conventions
for input and output that will be essentially the same (or at least
backward compatible) in all of the fancier versions that are to come.
We're given a matrix of 0s and 1s, whose columns represent ``items''
and whose rows represent ``options.'' Some of the items are called
``primary'' while the others are ``secondary.''
Every option contains a~1 for at least one primary item.
The problem is to find all subsets of the options whose sum is
(i)~{\it exactly\/}~1 for all primary items;
(ii)~{\it at most\/}~1 for all secondary items.
This matrix, which is typically very sparse, is specified on |stdin|
as follows:
\smallskip\item{$\bullet$} Each item has a symbolic name,
from one to eight characters long. Each of those characters can
be any nonblank ASCII code except for `\.{:}' and~`\.{\char"7C}'.
\smallskip\item{$\bullet$} The first line of input contains the
names of all primary items, separated by one or more spaces,
followed by `\.{\char"7C}', followed by the names of all other items.
(If all items are primary, the~`\.{\char"7C}' may be omitted.)
\smallskip\item{$\bullet$} The remaining lines represent the options,
by listing the items where 1~appears.
\smallskip\item{$\bullet$} Additionally, ``comment'' lines can be
interspersed anywhere in the input. Such lines, which begin with
`\.{\char"7C}', are ignored by this program, but they are often
useful within stored files.
\smallskip\noindent
Later versions of this program solve more general problems by
making further use of the reserved characters `\.{:}' and~`\.{\char"7C}'
to allow additional kinds of input.
For example, if we consider the matrix
$$\pmatrix{0&0&1&0&1&1&0\cr 1&0&0&1&0&0&1\cr 0&1&1&0&0&1&0\cr
1&0&0&1&0&0&0\cr 0&1&0&0&0&0&1\cr 0&0&0&1&1&0&1\cr}$$
which was (3) in my original paper, we can name the items
\.A, \.B, \.C, \.D, \.E, \.F,~\.G. Suppose the first five are
primary, and the latter two are secondary. That matrix can be
represented by the lines
$$
\vcenter{\halign{\tt#\cr
\char"7C\ A simple example\cr
A B C D E \char"7C\ F G\cr
C E F\cr
A D G\cr
B C F\cr
A D\cr
B G\cr
D E G\cr}}
$$
(and also in many other ways, because item names can be given in
any order, and so can the individual options). It has a unique solution,
consisting of the three options \.{A D} and \.{E F C} and \.{B G}.
@ After this program finds all solutions, it normally prints their total
number on |stderr|, together with statistics about how many
nodes were in the search tree, and how many ``updates'' were made.
The running time in ``mems'' is also reported, together with the approximate
number of bytes needed for data storage. One ``mem'' essentially means a
memory access to a 64-bit word.
(These totals don't include the time or space needed to parse the
input or to format the output.)
Here is the overall structure:
@d o mems++ /* count one mem */
@d oo mems+=2 /* count two mems */
@d ooo mems+=3 /* count three mems */
@d O "%" /* used for percent signs in format strings */
@d mod % /* used for percent signs denoting remainder in \CEE/ */
@d max_level 10000 /* at most this many options in a solution */
@d max_cols 100000 /* at most this many items */
@d max_nodes 25000000 /* at most this many nonzero elements in the matrix */
@d bufsize (9*max_cols+3) /* a buffer big enough to hold all item names */
@c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include "gb_flip.h"
typedef unsigned int uint; /* a convenient abbreviation */
typedef unsigned long long ullng; /* ditto */
@<Type definitions@>;
@<Global variables@>;
@<Subroutines@>;
main (int argc, char *argv[]) {
register int cc,i,j,k,p,pp,q,r,t,cur_node,best_itm;
@<Process the command line@>;
@<Input the item names@>;
@<Input the options@>;
if (vbose&show_basics)
@<Report the successful completion of the input phase@>;
if (vbose&show_tots)
@<Report the item totals@>;
imems=mems, mems=0;
@<Solve the problem@>;
done:@+if (vbose&show_tots)
@<Report the item totals@>;
if (vbose&show_profile) @<Print the profile@>;
if (vbose&show_max_deg)
fprintf(stderr,"The maximum branching degree was "O"d.\n",maxdeg);
if (vbose&show_basics) {
fprintf(stderr,"Altogether "O"llu solution"O"s, "O"llu+"O"llu mems,",
count,count==1?"":"s",imems,mems);
bytes=last_itm*sizeof(item)+last_node*sizeof(node)+maxl*sizeof(int);
fprintf(stderr," "O"llu updates, "O"llu bytes, "O"llu nodes,",
updates,bytes,nodes);
fprintf(stderr," ccost "O"lld%%.\n",
(200*cmems+mems)/(2*mems));
}
@<Close the files@>;
}
@ You can control the amount of output, as well as certain properties
of the algorithm, by specifying options on the command line:
\smallskip\item{$\bullet$}
`\.v$\langle\,$integer$\,\rangle$' enables or disables various kinds of verbose
output on |stderr|, given by binary codes such as |show_choices|;
\item{$\bullet$}
`\.m$\langle\,$integer$\,\rangle$' causes every $m$th solution
to be output (the default is \.{m0}, which merely counts them);
\item{$\bullet$}
`\.s$\langle\,$integer$\,\rangle$' causes the algorithm to make
random choices in key places (thus providing some variety, although
the solutions are by no means uniformly random), and it also
defines the seed for any random numbers that are used;
\item{$\bullet$}
`\.d$\langle\,$integer$\,\rangle$' sets |delta|, which causes periodic
state reports on |stderr| after the algorithm has performed approximately
|delta| mems since the previous report (default 10000000000);
\item{$\bullet$}
`\.c$\langle\,$positive integer$\,\rangle$' limits the levels on which
choices are shown during verbose tracing;
\item{$\bullet$}
`\.C$\langle\,$positive integer$\,\rangle$' limits the levels on which
choices are shown in the periodic state reports;
\item{$\bullet$}
`\.l$\langle\,$nonnegative integer$\,\rangle$' gives a {\it lower\/} limit,
relative to the maximum level so far achieved, to the levels on which
choices are shown during verbose tracing;
\item{$\bullet$}
`\.t$\langle\,$positive integer$\,\rangle$' causes the program to
stop after this many solutions have been found;
\item{$\bullet$}
`\.T$\langle\,$integer$\,\rangle$' sets |timeout| (which causes abrupt
termination if |mems>timeout| at the beginning of a level);
\item{$\bullet$}
`\.S$\langle\,$filename$\,\rangle$' to output a ``shape file'' that encodes
the search tree.
@d show_basics 1 /* |vbose| code for basic stats; this is the default */
@d show_choices 2 /* |vbose| code for backtrack logging */
@d show_details 4 /* |vbose| code for further commentary */
@d show_profile 128 /* |vbose| code to show the search tree profile */
@d show_full_state 256 /* |vbose| code for complete state reports */
@d show_tots 512 /* |vbose| code for reporting item totals at start and end */
@d show_warnings 1024 /* |vbose| code for reporting options without primaries */
@d show_max_deg 2048 /* |vbose| code for reporting maximum branching degree */
@<Glob...@>=
int random_seed=0; /* seed for the random words of |gb_rand| */
int randomizing; /* has `\.s' been specified? */
int vbose=show_basics+show_warnings; /* level of verbosity */
int spacing; /* solution $k$ is output if $k$ is a multiple of |spacing| */
int show_choices_max=1000000; /* above this level, |show_choices| is ignored */
int show_choices_gap=1000000; /* below level |maxl-show_choices_gap|,
|show_details| is ignored */
int show_levels_max=1000000; /* above this level, state reports stop */
int maxl=0; /* maximum level actually reached */
char buf[bufsize]; /* input buffer */
ullng count; /* solutions found so far */
ullng options; /* options seen so far */
ullng imems,mems,cmems,tmems; /* mem counts */
ullng updates; /* update counts */
ullng bytes; /* memory used by main data structures */
ullng nodes; /* total number of branch nodes initiated */
ullng thresh=10000000000; /* report when |mems| exceeds this, if |delta!=0| */
ullng delta=10000000000; /* report every |delta| or so mems */
ullng maxcount=0xffffffffffffffff; /* stop after finding this many solutions */
ullng timeout=0x1fffffffffffffff; /* give up after this many mems */
FILE *shape_file; /* file for optional output of search tree shape */
char *shape_name; /* its name */
int maxdeg; /* the largest branching degree seen so far */
@ If an option appears more than once on the command line, the first
appearance takes precedence.
@<Process the command line@>=
for (j=argc-1,k=0;j;j--) switch (argv[j][0]) {
case 'v': k|=(sscanf(argv[j]+1,""O"d",&vbose)-1);@+break;
case 'm': k|=(sscanf(argv[j]+1,""O"d",&spacing)-1);@+break;
case 's': k|=(sscanf(argv[j]+1,""O"d",&random_seed)-1),randomizing=1;@+break;
case 'd': k|=(sscanf(argv[j]+1,""O"lld",&delta)-1),thresh=delta;@+break;
case 'c': k|=(sscanf(argv[j]+1,""O"d",&show_choices_max)-1);@+break;
case 'C': k|=(sscanf(argv[j]+1,""O"d",&show_levels_max)-1);@+break;
case 'l': k|=(sscanf(argv[j]+1,""O"d",&show_choices_gap)-1);@+break;
case 't': k|=(sscanf(argv[j]+1,""O"lld",&maxcount)-1);@+break;
case 'T': k|=(sscanf(argv[j]+1,""O"lld",&timeout)-1);@+break;
case 'S': shape_name=argv[j]+1, shape_file=fopen(shape_name,"w");
if (!shape_file)
fprintf(stderr,"Sorry, I can't open file `"O"s' for writing!\n",
shape_name);
break;
default: k=1; /* unrecognized command-line option */
}
if (k) {
fprintf(stderr, "Usage: "O"s [v<n>] [m<n>] [s<n>] [d<n>]"
" [c<n>] [C<n>] [l<n>] [t<n>] [T<n>] [S<bar>] < foo.dlx\n",
argv[0]);
exit(-1);
}
if (randomizing) gb_init_rand(random_seed);
@ @<Close the files@>=
if (shape_file) fclose(shape_file);
@*Data structures.
Each item of the input matrix is represented by an \&{item} struct,
and each option is represented as a list of \&{node} structs. There's one
node for each nonzero entry in the matrix.
More precisely, the nodes of individual options appear sequentially,
with ``spacer'' nodes between them. The nodes are also
linked circularly with respect to each item, in doubly linked lists.
The item lists each include a header node, but the option lists do not.
Item header nodes are aligned with an \&{item} struct, which
contains further info about the item.
Each node contains three important fields. Two are the pointers |up|
and |down| of doubly linked lists, already mentioned.
The third points directly to the item containing the node.
A ``pointer'' is an array index, not a \CEE/ reference (because the latter
would occupy 64~bits and waste cache space). The |cl| array is for
\&{item} structs, and the |nd| array is for \&{node}s. I assume that both of
those arrays are small enough to be allocated statically. (Modifications
of this program could do dynamic allocation if needed.)
The header node corresponding to |cl[c]| is |nd[c]|.
We count one mem for a simultaneous access to the |up| and |down| fields.
I've added a |spare| field, so that each \&{node} occupies two
octabytes.
Although the item-list pointers are called |up| and |down|, they need not
correspond to actual positions of matrix entries. The elements of
each item list can appear in any order, so that one option
needn't be consistently ``above'' or ``below'' another. Indeed, when
|randomizing| is set, we intentionally scramble each item list.
This program doesn't change the |itm| fields after they've first been set up.
But the |up| and |down| fields will be changed frequently, although preserving
relative order.
Exception: In the node |nd[c]| that is the header for the list of
item~|c|, we use the |itm| field to hold the {\it length\/} of that
list (excluding the header node itself).
We also might use its |spare| field for special purposes.
The alternative names |len| for |itm| and |aux| for |spare|
are used in the code so that this nonstandard semantics will be more clear.
A {\it spacer\/} node has |itm<=0|. Its |up| field points to the start
of the preceding option; its |down| field points to the end of the following option.
Thus it's easy to traverse an option circularly, in either direction.
If all options have length |m|, we can do without the spacers by simply
working modulo~|m|. But the majority of my applications have options of
variable length, so I've decided not to use that trick.
[{\it Historical note:\/} An earlier version of this program, {\mc DLX0},
was almost identical to this one except that it used doubly linked lists
for the options as well as for the items. Thus it had two additional
fields, |left| and |right|, in each node. When I wrote {\mc DLX1} I expected
it to be a big improvement, because I thought there would be fewer memory
accesses in all of the inner loops where options are being traversed. However,
I~failed to realize that the |itm| and |right| fields were both stored in the
same octabyte; hence the cost per node is the same---and
{\mc DLX1} actually performs a few {\it more\/} mems, as
it handles the spacer node transitions! This additional mem cost is
compensated by the smaller node size, hence greater likelihood of cache hits.
But the gain from pure sequential allocation wasn't as great as I'd hoped.]
@d len itm /* item list length (used in header nodes only) */
@d aux spare /* an auxiliary quantity (used in header nodes only) */
@<Type...@>=
typedef struct node_struct {
int up,down; /* predecessor and successor in item list */
int itm; /* the item containing this node */
int spare; /* padding, not used in {\mc DLX1} */
} node;
@ Each \&{item} struct contains three fields:
The |name| is the user-specified identifier;
|next| and |prev| point to adjacent items, when this
item is part of a doubly linked list.
As backtracking proceeds, nodes
will be deleted from item lists when their option has been hidden by
other options in the partial solution.
But when backtracking is complete, the data structures will be
restored to their original state.
We count one mem for a simultaneous access to the |prev| and |next| fields.
@<Type...@>=
typedef struct itm_struct {
char name[8]; /* symbolic identification of the item, for printing */
int prev,next; /* neighbors of this item */
} item;
@ @<Glob...@>=
node nd[max_nodes]; /* the master list of nodes */
int last_node; /* the first node in |nd| that's not yet used */
item cl[max_cols+2]; /* the master list of items */
int second=max_cols; /* boundary between primary and secondary items */
int last_itm; /* the first item in |cl| that's not yet used */
@ One |item| struct is called the root. It serves as the head of the
list of items that need to be covered, and is identifiable by the fact
that its |name| is empty.
@d root 0 /* |cl[root]| is the gateway to the unsettled items */
@ An option is identified not by name but by the names of the items it contains.
Here is a routine that prints an option, given a pointer to any of its
nodes. It also prints the position of the option in its item list.
@<Sub...@>=
void print_option(int p,FILE *stream) {
register int k,q;
if (p<last_itm || p>=last_node || nd[p].itm<=0) {
fprintf(stderr,"Illegal option "O"d!\n",p);
return;
}
for (q=p;;) {
fprintf(stream," "O".8s",cl[nd[q].itm].name);
q++;
if (nd[q].itm<=0) q=nd[q].up; /* |-nd[q].itm| is actually the option number */
if (q==p) break;
}
for (q=nd[nd[p].itm].down,k=1;q!=p;k++) {
if (q==nd[p].itm) {
fprintf(stream," (?)\n");@+return; /* option not in its item list! */
}@+else q=nd[q].down;
}
fprintf(stream," ("O"d of "O"d)\n",k,nd[nd[p].itm].len);
}
@#
void prow(int p) {
print_option(p,stderr);
}
@ When I'm debugging, I might want to look at one of the current item lists.
@<Sub...@>=
void print_itm(int c) {
register int p;
if (c<root || c>=last_itm) {
fprintf(stderr,"Illegal item "O"d!\n",c);
return;
}
if (c<second)
fprintf(stderr,"Item "O".8s, length "O"d, neighbors "O".8s and "O".8s:\n",
cl[c].name,nd[c].len,cl[cl[c].prev].name,cl[cl[c].next].name);
else fprintf(stderr,"Item "O".8s, length "O"d:\n",cl[c].name,nd[c].len);
for (p=nd[c].down;p>=last_itm;p=nd[p].down) prow(p);
}
@ Speaking of debugging, here's a routine to check if redundant parts of our
data structure have gone awry.
@d sanity_checking 0 /* set this to 1 if you suspect a bug */
@<Sub...@>=
void sanity(void) {
register int k,p,q,pp,qq,t;
for (q=root,p=cl[q].next;;q=p,p=cl[p].next) {
if (cl[p].prev!=q) fprintf(stderr,"Bad prev field at itm "O".8s!\n",
cl[p].name);
if (p==root) break;
@<Check item |p|@>;
}
}
@ @<Check item |p|@>=
for (qq=p,pp=nd[qq].down,k=0;;qq=pp,pp=nd[pp].down,k++) {
if (nd[pp].up!=qq) fprintf(stderr,"Bad up field at node "O"d!\n",pp);
if (pp==p) break;
if (nd[pp].itm!=p) fprintf(stderr,"Bad itm field at node "O"d!\n",pp);
}
if (nd[p].len!=k) fprintf(stderr,"Bad len field in item "O".8s!\n",
cl[p].name);
@*Inputting the matrix. Brute force is the rule in this part of the code,
whose goal is to parse and store the input data and to check its validity.
@d panic(m) {@+fprintf(stderr,""O"s!\n"O"d: "O".99s\n",m,p,buf);@+exit(-666);@+}
@<Input the item names@>=
if (max_nodes<=2*max_cols) {
fprintf(stderr,"Recompile me: max_nodes must exceed twice max_cols!\n");
exit(-999);
} /* every item will want a header node and at least one other node */
while (1) {
if (!fgets(buf,bufsize,stdin)) break;
if (o,buf[p=strlen(buf)-1]!='\n') panic("Input line way too long");
for (p=0;o,isspace(buf[p]);p++) ;
if (buf[p]=='|' || !buf[p]) continue; /* bypass comment or blank line */
last_itm=1;
break;
}
if (!last_itm) panic("No items");
for (;o,buf[p];) {
for (j=0;j<8 && (o,!isspace(buf[p+j]));j++) {
if (buf[p+j]==':' || buf[p+j]=='|')
panic("Illegal character in item name");
o,cl[last_itm].name[j]=buf[p+j];
}
if (j==8 && !isspace(buf[p+j])) panic("Item name too long");
@<Check for duplicate item name@>;
@<Initialize |last_itm| to a new item with an empty list@>;
for (p+=j+1;o,isspace(buf[p]);p++) ;
if (buf[p]=='|') {
if (second!=max_cols) panic("Item name line contains | twice");
second=last_itm;
for (p++;o,isspace(buf[p]);p++) ;
}
}
if (second==max_cols) second=last_itm;
oo,cl[last_itm].prev=last_itm-1, cl[last_itm-1].next=last_itm;
oo,cl[second].prev=last_itm,cl[last_itm].next=second;
/* this sequence works properly whether or not |second=last_itm| */
oo,cl[root].prev=second-1, cl[second-1].next=root;
last_node=last_itm; /* reserve all the header nodes and the first spacer */
/* we have |nd[last_node].itm=0| in the first spacer */
@ @<Check for duplicate item name@>=
for (k=1;o,strncmp(cl[k].name,cl[last_itm].name,8);k++) ;
if (k<last_itm) panic("Duplicate item name");
@ @<Initialize |last_itm| to a new item with an empty list@>=
if (last_itm>max_cols) panic("Too many items");
oo,cl[last_itm-1].next=last_itm,cl[last_itm].prev=last_itm-1;
/* |nd[last_itm].len=0| */
o,nd[last_itm].up=nd[last_itm].down=last_itm;
last_itm++;
@ I'm putting the option number into the spacer that follows it, as a
possible debugging aid. But the program doesn't currently use that information.
@<Input the options@>=
while (1) {
if (!fgets(buf,bufsize,stdin)) break;
if (o,buf[p=strlen(buf)-1]!='\n') panic("Option line too long");
for (p=0;o,isspace(buf[p]);p++) ;
if (buf[p]=='|' || !buf[p]) continue; /* bypass comment or blank line */
i=last_node; /* remember the spacer at the left of this option */
for (pp=0;buf[p];) {
for (j=0;j<8 && (o,!isspace(buf[p+j]));j++)
o,cl[last_itm].name[j]=buf[p+j];
if (j==8 && !isspace(buf[p+j])) panic("Item name too long");
if (j<8) o,cl[last_itm].name[j]='\0';
@<Create a node for the item named in |buf[p]|@>;
for (p+=j+1;o,isspace(buf[p]);p++) ;
}
if (!pp) {
if (vbose&show_warnings)
fprintf(stderr,"Option ignored (no primary items): "O"s",buf);
while (last_node>i) {
@<Remove |last_node| from its item list@>;
last_node--;
}
}@+else {
o,nd[i].down=last_node;
last_node++; /* create the next spacer */
if (last_node==max_nodes) panic("Too many nodes");
options++;
o,nd[last_node].up=i+1;
o,nd[last_node].itm=-options;
}
}
@ @<Create a node for the item named in |buf[p]|@>=
for (k=0;o,strncmp(cl[k].name,cl[last_itm].name,8);k++) ;
if (k==last_itm) panic("Unknown item name");
if (o,nd[k].aux>=i) panic("Duplicate item name in this option");
last_node++;
if (last_node==max_nodes) panic("Too many nodes");
o,nd[last_node].itm=k;
if (k<second) pp=1;
o,t=nd[k].len+1;
@<Insert node |last_node| into the list for item |k|@>;
@ Insertion of a new node is simple, unless we're randomizing.
In the latter case, we want to put the node into a random position
of the list.
We store the position of the new node into |nd[k].aux|, so that
the test for duplicate items above will be correct.
As in other programs developed for TAOCP, I assume that four mems are
consumed when 31 random bits are being generated by any of the {\mc GB\_FLIP}
routines.
@<Insert node |last_node| into the list for item |k|@>=
o,nd[k].len=t; /* store the new length of the list */
nd[k].aux=last_node; /* no mem charge for |aux| after |len| */
if (!randomizing) {
o,r=nd[k].up; /* the ``bottom'' node of the item list */
ooo,nd[r].down=nd[k].up=last_node,nd[last_node].up=r,nd[last_node].down=k;
}@+else {
mems+=4,t=gb_unif_rand(t); /* choose a random number of nodes to skip past */
for (o,r=k;t;o,r=nd[r].down,t--) ;
ooo,q=nd[r].up,nd[q].down=nd[r].up=last_node;
o,nd[last_node].up=q,nd[last_node].down=r;
}
@ @<Remove |last_node| from its item list@>=
o,k=nd[last_node].itm;
oo,nd[k].len--,nd[k].aux=i-1;
o,q=nd[last_node].up,r=nd[last_node].down;
oo,nd[q].down=r,nd[r].up=q;
@ @<Report the successful completion of the input phase@>=
fprintf(stderr,
"("O"lld options, "O"d+"O"d items, "O"d entries successfully read)\n",
options,second-1,last_itm-second,last_node-last_itm);
@ The item lengths after input should agree with the item lengths
after this program has finished. I print them (on request), in order to
provide some reassurance that the algorithm isn't badly screwed up.
@<Report the item totals@>=
{
fprintf(stderr,"Item totals:");
for (k=1;k<last_itm;k++) {
if (k==second) fprintf(stderr," |");
fprintf(stderr," "O"d",nd[k].len);
}
fprintf(stderr,"\n");
}
@*The dancing.
Our strategy for generating all exact covers will be to repeatedly
choose always the item that appears to be hardest to cover, namely the
item with shortest list, from all items that still need to be covered.
And we explore all possibilities via depth-first search.
The neat part of this algorithm is the way the lists are maintained.
Depth-first search means last-in-first-out maintenance of data structures;
and it turns out that we need no auxiliary tables to undelete elements from
lists when backing up. The nodes removed from doubly linked lists remember
their former neighbors, because we do no garbage collection.
The basic operation is ``covering an item.'' This means removing it
from the list of items needing to be covered, and ``hiding'' its
options: removing nodes from other lists whenever they belong to an option of
a node in this item's list.
@<Solve the problem@>=
level=0;
forward: nodes++;
if (vbose&show_profile) profile[level]++;
if (sanity_checking) sanity();
@<Do special things if enough |mems| have accumulated@>;
@<Set |best_itm| to the best item for branching@>;
cover(best_itm);
oo,cur_node=choice[level]=nd[best_itm].down;
advance:@+if (cur_node==best_itm) goto backup;
if ((vbose&show_choices) && level<show_choices_max) {
fprintf(stderr,"L"O"d:",level);
print_option(cur_node,stderr);
}
@<Cover all other items of |cur_node|@>;
if (o,cl[root].next==root) @<Visit a solution and |goto recover|@>;
if (++level>maxl) {
if (level>=max_level) {
fprintf(stderr,"Too many levels!\n");
exit(-4);
}
maxl=level;
}
goto forward;
backup: uncover(best_itm);
if (level==0) goto done;
level--;
oo,cur_node=choice[level],best_itm=nd[cur_node].itm;
recover: @<Uncover all other items of |cur_node|@>;
oo,cur_node=choice[level]=nd[cur_node].down;@+goto advance;
@ @<Glob...@>=
int level; /* number of choices in current partial solution */
int choice[max_level]; /* the node chosen on each level */
ullng profile[max_level]; /* number of search tree nodes on each level */
@ @<Do special things if enough |mems| have accumulated@>=
if (delta && (mems>=thresh)) {
thresh+=delta;
if (vbose&show_full_state) print_state();
else print_progress();
}
if (mems>=timeout) {
fprintf(stderr,"TIMEOUT!\n");@+goto done;
}
@ When an option is hidden, it leaves all lists except the list of the
item that is being covered. Thus a node is never removed from a list
twice.
Note: I could have saved some mems in this routine, and in similar
routines below, by not updating the |len| fields of secondary items.
But I chose not to make such an optimization because it might well be
misleading: The insertion of a mem-free new branch `|if (cc<second)|'
can be costly since it makes hardware branch prediction less effective.
Furthermore those |len| fields are in item header nodes, which tend
to remain in cache memory where they're readily accessible.
@<Sub...@>=
void cover(int c) {
register int cc,l,r,rr,nn,uu,dd,t;
o,l=cl[c].prev,r=cl[c].next;
oo,cl[l].next=r,cl[r].prev=l;
updates++;
for (o,rr=nd[c].down;rr>=last_itm;o,rr=nd[rr].down)
for (nn=rr+1;nn!=rr;) {
o,uu=nd[nn].up,dd=nd[nn].down;
o,cc=nd[nn].itm;
if (cc<=0) {
nn=uu;
continue;
}
oo,nd[uu].down=dd,nd[dd].up=uu;
updates++;
o,t=nd[cc].len-1;
o,nd[cc].len=t;
nn++;
}
}
@ I used to think that it was important to uncover an item by
processing its options from bottom to top, since covering was done
from top to bottom. But while writing this
program I realized that, amazingly, no harm is done if the
options are processed again in the same order. So I'll go downward again,
just to prove the point. Whether we go up or down, the pointers
execute an exquisitely choreo\-graphed dance that returns them almost
magically to their former state.
@<Subroutines@>=
void uncover(int c) {
register int cc,l,r,rr,nn,uu,dd,t;
for (o,rr=nd[c].down;rr>=last_itm;o,rr=nd[rr].down)
for (nn=rr+1;nn!=rr;) {
o,uu=nd[nn].up,dd=nd[nn].down;
o,cc=nd[nn].itm;
if (cc<=0) {
nn=uu;
continue;
}
oo,nd[uu].down=nd[dd].up=nn;
o,t=nd[cc].len+1;
o,nd[cc].len=t;
nn++;
}
o,l=cl[c].prev,r=cl[c].next;
oo,cl[l].next=cl[r].prev=c;
}
@ @<Cover all other items of |cur_node|@>=
for (pp=cur_node+1;pp!=cur_node;) {
o,cc=nd[pp].itm;
if (cc<=0) o,pp=nd[pp].up;
else cover(cc),pp++;
}
@ When I learned that the covering of individual items can be done safely in
various orders, I~almost convinced myself that I'd be able
to blithely ignore the ordering---I could apparently
undo the covering of item $a$ then $b$ by uncovering $a$ first.
However, that argument is fallacious: When $a$ is uncovered, it
can resuscitate elements in item~$b$ that would mess up the
uncovering of~$b$. The choreography is delicate indeed.
(Incidentally, the |cover| and |uncover| routines both went to the right.
That was okay. But we must then go left here.)
@<Uncover all other items of |cur_node|@>=
for (pp=cur_node-1;pp!=cur_node;) {
o,cc=nd[pp].itm;
if (cc<=0) o,pp=nd[pp].down;
else uncover(cc),pp--;
}
@ The ``best item'' is considered to be an item that minimizes the
number of remaining choices. If there are several candidates, we
choose the leftmost --- unless we're randomizing, in which case we
select one of them at random.
@<Set |best_itm| to the best item for branching@>=
tmems=mems,t=max_nodes;
if ((vbose&show_details) &&
level<show_choices_max && level>=maxl-show_choices_gap)
fprintf(stderr,"Level "O"d:",level);
for (o,k=cl[root].next;t&&k!=root;o,k=cl[k].next) {
if ((vbose&show_details) &&
level<show_choices_max && level>=maxl-show_choices_gap)
fprintf(stderr," "O".8s("O"d)",cl[k].name,nd[k].len);
if (o,nd[k].len<=t) {
if (nd[k].len<t) best_itm=k,t=nd[k].len,p=1;
else {
p++; /* this many items achieve the min */
if (randomizing && (mems+=4,!gb_unif_rand(p))) best_itm=k;
}
}
}
if ((vbose&show_details) &&
level<show_choices_max && level>=maxl-show_choices_gap)
fprintf(stderr," branching on "O".8s("O"d)\n",cl[best_itm].name,t);
if (t>maxdeg) maxdeg=t;
if (shape_file) {
fprintf(shape_file,""O"d "O".8s\n",t,cl[best_itm].name);
fflush(shape_file);
}
cmems+=mems-tmems;
@ @<Visit a solution and |goto recover|@>=
{
nodes++; /* a solution is a special node, see 7.2.2--(4) */
if (level+1>maxl) {
if (level+1>=max_level) {
fprintf(stderr,"Too many levels!\n");
exit(-5);
}
maxl=level+1;
}
if (vbose&show_profile) profile[level+1]++;
if (shape_file) {
fprintf(shape_file,"sol\n");@+fflush(shape_file);
}
@<Record solution and |goto recover|@>;
}
@ @<Record solution and |goto recover|@>=
{
count++;
if (spacing && (count mod spacing==0)) {
printf(""O"lld:\n",count);
for (k=0;k<=level;k++) print_option(choice[k],stdout);
fflush(stdout);
}
if (count>=maxcount) goto done;
goto recover;
}
@ @<Sub...@>=
void print_state(void) {
register int l;
fprintf(stderr,"Current state (level "O"d):\n",level);
for (l=0;l<level;l++) {
print_option(choice[l],stderr);
if (l>=show_levels_max) {
fprintf(stderr," ...\n");
break;
}
}
fprintf(stderr," "O"lld solutions, "O"lld mems, and max level "O"d so far.\n",
count,mems,maxl);
}
@ During a long run, it's helpful to have some way to measure progress.
The following routine prints a string that indicates roughly where we
are in the search tree. The string consists of character pairs, separated
by blanks, where each character pair represents a branch of the search
tree. When a node has $d$ descendants and we are working on the $k$th,
the two characters respectively represent $k$ and~$d$ in a simple code;
namely, the values 0, 1, \dots, 61 are denoted by
$$\.0,\ \.1,\ \dots,\ \.9,\ \.a,\ \.b,\ \dots,\ \.z,\ \.A,\ \.B,\ \dots,\.Z.$$
All values greater than 61 are shown as `\.*'. Notice that as computation
proceeds, this string will increase lexicographically.
Following that string, a fractional estimate of total progress is computed,
based on the na{\"\i}ve assumption that the search tree has a uniform
branching structure. If the tree consists
of a single node, this estimate is~.5; otherwise, if the first choice
is `$k$ of~$d$', the estimate is $(k-1)/d$ plus $1/d$ times the
recursively evaluated estimate for the $k$th subtree. (This estimate
might obviously be very misleading, in some cases, but at least it
tends to grow monotonically.)
@<Sub...@>=
void print_progress(void) {
register int l,k,d,c,p;
register double f,fd;
fprintf(stderr," after "O"lld mems: "O"lld sols,",mems,count);
for (f=0.0,fd=1.0,l=0;l<level;l++) {
c=nd[choice[l]].itm,d=nd[c].len;
for (k=1,p=nd[c].down;p!=choice[l];k++,p=nd[p].down) ;
fd*=d,f+=(k-1)/fd; /* choice |l| is |k| of |d| */
fprintf(stderr," "O"c"O"c",
k<10? '0'+k: k<36? 'a'+k-10: k<62? 'A'+k-36: '*',
d<10? '0'+d: d<36? 'a'+d-10: d<62? 'A'+d-36: '*');
if (l>=show_levels_max) {
fprintf(stderr,"...");
break;
}
}
fprintf(stderr," "O".5f\n",f+0.5/fd);
}
@ @<Print the profile@>=
{
fprintf(stderr,"Profile:\n");
for (level=0;level<=maxl;level++)
fprintf(stderr,""O"3d: "O"lld\n",
level,profile[level]);
}
@*Index.