UVa Problem Solution: 10267 - Graphical Editor

The flood fill algorithm is somewhat tricky to implement. I use a iterative method instead of recursive flooding. I first scan through the west and east points to the current point until a different colored one is encountered. Then, the points between the west and east boundary are colored. Finally, the very north and south points to these points that with the same color are putted in a queue. Repeat these steps for each point in the queue would do the job.

When coloring the points horizontally,  points in the queue may be colored as well. Thus, if a point is already colored when it is dequeued, there is no need to check for it at all. I missed this point at first, thus making my program run more slowly than the recursive version.

I also implement a recursive version of the flood fill algorithm. Many others claim that the recursive would make the stack overflow or take a longer time to run. To my deep surprise, though I have tried my best to optimize the iterative version, the recursive version gets a better run time and does not cause a run time error on the judge.

Code:

1. /*************************************************************************
2.  * Copyright (C) 2008 by liukaipeng                                      *
3.  * liukaipeng at gmail dot com                                           *
4.  *************************************************************************/
6. /* @JUDGE_ID 00000 10267 C "Graphical Editor" */
8. #include <stdio.h>
9. #include <stdlib.h>
10. #include <string.h>
11. #include <strings.h>
13. #define MAX 252
15. /*------------------------------------------------*
16.  * Queue Implementation                           *
17.  *------------------------------------------------*/
19. #define QUEUE_MAX (MAX*MAX)
21. typedef struct point
22. {
23.   int x;
24.   int y;
25. } point;
27. typedef point elemtype;
29. typedef struct queue {
30.   elemtype data[QUEUE_MAX];
31.   int front;
32.   int tail;
33.   int size;
34. } queue;
36. void init(queue *q) {
37.   q->front = 0;
38.   q->tail = 0;
39.   q->size = 0;
40. }
42. int empty(queue *q) {
43.   return q->size == 0;
44. }
46. void enqueue(queue *q, elemtype e) {
47.   q->data[q->tail] = e;
48.   ++q->size;
49.   ++q->tail;
50.   q->tail %= QUEUE_MAX;
51. }
53. elemtype dequeue(queue *q) {
54.   elemtype e = q->data[q->front];
55.   --q->size;
56.   ++q->front;
57.   q->front %= QUEUE_MAX;
58.   return e;
59. }
61. /*------------------------------------------------*/
63. char bitmap[MAX][MAX];
64. int xmax, ymax;
66. void handle_C(char param[])
67. {
68.   int y;
69.   memset(bitmap, 'O', MAX*MAX);
70.   for (y = 1; y <= ymax; ++y)
71.     bitmap[y][xmax+1] = '/0';
72. }
74. void handle_I(char param[])
75. {
76.   sscanf(param, "%d %d", &xmax, &ymax);
77.   handle_C(param);
78. }
80. void handle_L(char param[])
81. {
82.   int x, y, c;
83.   sscanf(param, "%d %d %c", &x, &y, &c);
84.   bitmap[y][x] = c;
85. }
87. void handle_V(char param[])
88. {
89.   int x, y1, y2, c;
90.   sscanf(param, "%d %d %d %c", &x, &y1, &y2, &c);
91.   if (y1 > y2) 
92.     y1 ^= y2, y2^= y1, y1 ^= y2;
93.   for (; y1 <= y2; ++y1)
94.     bitmap[y1][x] = c;
95. }
97. void handle_H(char param[])
98. {
99.   int x1, x2, y, c;
100.   sscanf(param, "%d %d %d %c", &x1, &x2, &y, &c);
101.   if (x1 > x2)
102.     x1 ^= x2, x2^= x1, x1 ^= x2;
103.   memset(bitmap[y]+x1, c, x2-x1+1);
104. }
106. void handle_K(char param[])
107. {
108.   int x1, y1, x2, y2, c;
109.   sscanf(param, "%d %d %d %d %c", &x1, &y1, &x2, &y2, &c);
110.   if (x1 > x2)
111.     x1 ^= x2, x2^= x1, x1 ^= x2;
112.   if (y1 > y2) 
113.     y1 ^= y2, y2^= y1, y1 ^= y2;
114.   for (; y1 <= y2; ++y1) 
115.     memset(bitmap[y1]+x1, c, x2-x1+1);
116. }
118. void flood(int x, int y, char o, char c)
119. {
120.   if (c == o)
121.     return;
122.   bitmap[y][x] = c;
123.   if (x-1 > 0 && bitmap[y][x-1] == o)
124.     flood(x-1, y, o, c);
125.   if (y-1 > 0 && bitmap[y-1][x] == o)
126.     flood(x, y-1, o, c);
127.   if (x+1 < xmax+1 && bitmap[y][x+1] == o)
128.     flood(x+1, y, o, c);
129.   if (y+1 < ymax+1 && bitmap[y+1][x] == o)
130.     flood(x, y+1, o, c);
131. } 
133. void handle_F(char param[])
134. {
135.   point p;
136.   queue q;
137.   char o, c;
138.   int w, e, n, s;
140.   sscanf(param, "%d %d %c", &p.x, &p.y, &c);
141.   o = bitmap[p.y][p.x];
142.   /*  flood(p.x, p.y, o, c);*/
143.   init(&q);
144.   enqueue(&q, p);
145.   while (!empty(&q)) {
146.     p = dequeue(&q);
147.     if (bitmap[p.y][p.x] == c)
148.       continue;
149.     for (w = p.x; w-1 > 0 && bitmap[p.y][w-1] == o; --w); 
150.     for (e = p.x; e+1 < xmax+1 && bitmap[p.y][e+1] == o; ++e);
151.     memset(bitmap[p.y]+w, c, e-w+1);
152.     n = p.y-1, s = p.y+1;
153.     if (n > 0)
154.       for (p.y = n, p.x = w; p.x <= e; ++p.x)
155.         if (bitmap[p.y][p.x] == o)
156.           enqueue(&q, p);
157.     if (s < ymax+1)
158.       for (p.y = s, p.x = w; p.x <= e; ++p.x)
159.         if (bitmap[p.y][p.x] == o)
160.           enqueue(&q, p);
161.   }
162. }
164. void handle_S(char param[])
165. {
166.   printf("%s/n", param);
167.   int x, y;
168.   for (y = 1; y <= ymax; ++y) {
169.     puts(bitmap[y]+1);
170.   }
171. }
173. int main(int argc, char *argv[])
174. {
175. #ifndef ONLINE_JUDGE
176.   char in[256];
177.   char out[256];
178.   strcpy(in, argv[0]);
179.   strcat(in, ".in");
180.   freopen(in, "r", stdin);
181.   strcpy(out, argv[0]);
182.   strcat(out, ".out");
183.   freopen(out, "w", stdout);
184. #endif
186.   char buf[1024];
187.   for (scanf("%[^/n]/n", buf); buf[0] != 'X'; scanf("%[^/n]/n", buf)) {
188.     switch(buf[0]) {
189.     case 'I':
190.       handle_I(buf+2);
191.       break;
192.     case 'C':
193.       handle_C(buf+2);
194.       break;
195.     case 'L':
196.       handle_L(buf+2);
197.       break;
198.     case 'V':
199.       handle_V(buf+2);
200.       break;
201.     case 'H':
202.       handle_H(buf+2);
203.       break;
204.     case 'K':
205.       handle_K(buf+2);
206.       break;
207.     case 'F':
208.       handle_F(buf+2);
209.       break;
210.     case 'S':
211.       handle_S(buf+2);
212.       break;
213.     case 'X':
214.       break;
215.     default:
216.       break;
217.     }
218.   }
219.   return 0;
220. }
221.