#include "artclass.h"
#include<bits/stdc++.h>
using namespace std;
struct pixel
{
double r, g, b;
pixel(double _r = 0.0, double _g = 0.0, double _b = 0.0)
{
r = _r;
g = _g;
b = _b;
}
};
//pixel avg_st[5];
pixel image[510][510];
int h, w;
pixel avg_dev[5];
void init()
{
// avg_st[1] = pixel(189.852, 166.758, 147.354);
// avg_st[2] = pixel(102.854, 98.1113, 64.2302);
// avg_st[3] = pixel(140.586, 128.459, 107.031);
// avg_st[4] = pixel(158.108, 89.7725, 55.3149);
avg_dev[1] = pixel(2288.99, 2322.12, 2839.83);
avg_dev[2] = pixel(749.693, 627.756, 424.602);
avg_dev[3] = pixel(2897.08, 2767.16, 2537.68);
avg_dev[4] = pixel(169.811, 143.836, 248.756);
}
double penalty[5];
pixel average_pixel(int stx, int sty, int enx, int eny)
{
double sr = 0.0;
double sg = 0.0;
double sb = 0.0;
for (int i = stx; i <= enx; i ++)
for (int j = sty; j <= eny; j ++)
{
sr += image[i][j].r;
sg += image[i][j].g;
sb += image[i][j].b;
}
double cnt = (enx - stx + 1) * (eny - sty + 1);
sr /= cnt;
sg /= cnt;
sb /= cnt;
return pixel(sr, sg, sb);
}
pixel region_deviation(int stx, int sty, int enx, int eny)
{
pixel avg = average_pixel(stx, sty, enx, eny);
double dr = 0.0;
double dg = 0.0;
double db = 0.0;
for (int i = stx; i <= enx; i ++)
{
for (int j = sty; j <= eny; j ++)
{
double delta_red = abs(image[i][j].r - avg.r);
double delta_green = abs(image[i][j].g - avg.g);
double delta_blue = abs(image[i][j].b - avg.b);
delta_red = delta_red * delta_red;
delta_green = delta_green * delta_green;
delta_blue = delta_blue * delta_blue;
dr += delta_red;
dg += delta_green;
db += delta_blue;
}
}
double cnt = (enx - stx + 1) * (eny - sty + 1);
dr /= cnt;
dg /= cnt;
db /= cnt;
return pixel(dr, dg, db);
}
const int block = 20;
pixel get_average_block_deviation()
{
pixel avg;
double cnt = 0.0;
for (int i = 0; i < h; i += block)
for (int j = 0; j < w; j += block)
{
int stx = i, sty = j;
int enx = min(h, i + block) - 1;
int eny = min(w, j + block) - 1;
pixel cur = region_deviation(stx, sty, enx, eny);
for (int s = 1; s <= 4; s ++)
{
double dr = abs(cur.r - avg_dev[s].r);
double dg = abs(cur.g - avg_dev[s].g);
double db = abs(cur.b - avg_dev[s].b);
dr = dr * dr;
dg = dg * dg;
db = db * db;
penalty[s] += dr + dg + db;
}
avg.r += cur.r;
avg.g += cur.g;
avg.b += cur.b;
cnt ++;
}
avg.r /= cnt;
avg.g /= cnt;
avg.b /= cnt;
return avg;
//cout << "average deviation" << endl;
//cout << avg.r << " " << avg.g << " " << avg.b << endl;
}
int style(int H, int W, int R[500][500], int G[500][500], int B[500][500])
{
init();
h = H;
w = W;
for (int i = 0; i < h; i ++)
for (int j = 0; j < w; j ++)
{
image[i][j] = pixel(R[i][j], G[i][j], B[i][j]);
}
pixel dev = get_average_block_deviation();
int mx = 1;
for (int i = 1; i <= 4; i ++)
{
if (penalty[i] < penalty[mx])
mx = i;
}
return mx;
}
Compilation message
artclass.cpp: In function 'int style(int, int, int (*)[500], int (*)[500], int (*)[500])':
artclass.cpp:144:11: warning: variable 'dev' set but not used [-Wunused-but-set-variable]
144 | pixel dev = get_average_block_deviation();
| ^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
9308 KB |
Output is correct |
2 |
Correct |
38 ms |
9304 KB |
Output is correct |
3 |
Correct |
36 ms |
9456 KB |
Output is correct |
4 |
Correct |
40 ms |
9472 KB |
Output is correct |
5 |
Incorrect |
40 ms |
9304 KB |
Output isn't correct |
6 |
Correct |
41 ms |
9304 KB |
Output is correct |
7 |
Correct |
34 ms |
9308 KB |
Output is correct |
8 |
Correct |
28 ms |
9308 KB |
Output is correct |
9 |
Correct |
35 ms |
9308 KB |
Output is correct |
10 |
Incorrect |
26 ms |
9304 KB |
Output isn't correct |
11 |
Correct |
38 ms |
9304 KB |
Output is correct |
12 |
Incorrect |
54 ms |
9308 KB |
Output isn't correct |
13 |
Correct |
24 ms |
9484 KB |
Output is correct |
14 |
Correct |
26 ms |
9400 KB |
Output is correct |
15 |
Incorrect |
27 ms |
9308 KB |
Output isn't correct |
16 |
Correct |
34 ms |
9308 KB |
Output is correct |
17 |
Correct |
34 ms |
9308 KB |
Output is correct |
18 |
Correct |
34 ms |
9308 KB |
Output is correct |
19 |
Correct |
35 ms |
9304 KB |
Output is correct |
20 |
Correct |
28 ms |
9308 KB |
Output is correct |
21 |
Correct |
38 ms |
9456 KB |
Output is correct |
22 |
Correct |
32 ms |
9456 KB |
Output is correct |
23 |
Incorrect |
23 ms |
9308 KB |
Output isn't correct |
24 |
Correct |
43 ms |
9400 KB |
Output is correct |
25 |
Incorrect |
42 ms |
9456 KB |
Output isn't correct |
26 |
Correct |
34 ms |
9304 KB |
Output is correct |
27 |
Correct |
25 ms |
9556 KB |
Output is correct |
28 |
Correct |
41 ms |
9308 KB |
Output is correct |
29 |
Correct |
39 ms |
9308 KB |
Output is correct |
30 |
Correct |
25 ms |
9308 KB |
Output is correct |
31 |
Correct |
13 ms |
9304 KB |
Output is correct |
32 |
Correct |
37 ms |
9304 KB |
Output is correct |
33 |
Correct |
31 ms |
9308 KB |
Output is correct |
34 |
Correct |
25 ms |
9308 KB |
Output is correct |
35 |
Incorrect |
35 ms |
9308 KB |
Output isn't correct |
36 |
Correct |
33 ms |
9308 KB |
Output is correct |
37 |
Incorrect |
43 ms |
9308 KB |
Output isn't correct |
38 |
Correct |
33 ms |
9308 KB |
Output is correct |
39 |
Correct |
37 ms |
9308 KB |
Output is correct |
40 |
Correct |
31 ms |
9304 KB |
Output is correct |
41 |
Correct |
41 ms |
9452 KB |
Output is correct |
42 |
Correct |
29 ms |
9308 KB |
Output is correct |
43 |
Correct |
39 ms |
9480 KB |
Output is correct |
44 |
Correct |
36 ms |
9304 KB |
Output is correct |
45 |
Correct |
15 ms |
9308 KB |
Output is correct |
46 |
Incorrect |
33 ms |
9308 KB |
Output isn't correct |
47 |
Incorrect |
44 ms |
9300 KB |
Output isn't correct |
48 |
Incorrect |
43 ms |
9560 KB |
Output isn't correct |
49 |
Correct |
37 ms |
9472 KB |
Output is correct |
50 |
Incorrect |
34 ms |
9308 KB |
Output isn't correct |
51 |
Correct |
26 ms |
9308 KB |
Output is correct |
52 |
Correct |
35 ms |
9308 KB |
Output is correct |
53 |
Incorrect |
27 ms |
9308 KB |
Output isn't correct |
54 |
Correct |
30 ms |
9308 KB |
Output is correct |
55 |
Incorrect |
41 ms |
9460 KB |
Output isn't correct |
56 |
Correct |
37 ms |
9460 KB |
Output is correct |
57 |
Correct |
37 ms |
9308 KB |
Output is correct |
58 |
Incorrect |
23 ms |
9308 KB |
Output isn't correct |
59 |
Incorrect |
33 ms |
9376 KB |
Output isn't correct |
60 |
Correct |
18 ms |
9304 KB |
Output is correct |
61 |
Correct |
39 ms |
9456 KB |
Output is correct |
62 |
Correct |
34 ms |
9308 KB |
Output is correct |
63 |
Correct |
29 ms |
9308 KB |
Output is correct |
64 |
Incorrect |
35 ms |
9308 KB |
Output isn't correct |
65 |
Correct |
27 ms |
9308 KB |
Output is correct |
66 |
Correct |
28 ms |
9728 KB |
Output is correct |
67 |
Correct |
35 ms |
9308 KB |
Output is correct |
68 |
Correct |
29 ms |
9304 KB |
Output is correct |
69 |
Correct |
32 ms |
9308 KB |
Output is correct |
70 |
Correct |
34 ms |
9456 KB |
Output is correct |
71 |
Correct |
37 ms |
9476 KB |
Output is correct |
72 |
Incorrect |
37 ms |
9308 KB |
Output isn't correct |
73 |
Correct |
35 ms |
9308 KB |
Output is correct |
74 |
Correct |
32 ms |
9308 KB |
Output is correct |
75 |
Correct |
34 ms |
9308 KB |
Output is correct |
76 |
Correct |
33 ms |
9308 KB |
Output is correct |
77 |
Incorrect |
28 ms |
9308 KB |
Output isn't correct |
78 |
Incorrect |
41 ms |
9304 KB |
Output isn't correct |
79 |
Correct |
39 ms |
9308 KB |
Output is correct |
80 |
Incorrect |
46 ms |
9308 KB |
Output isn't correct |
81 |
Correct |
32 ms |
9308 KB |
Output is correct |
82 |
Correct |
34 ms |
9304 KB |
Output is correct |
83 |
Correct |
30 ms |
9308 KB |
Output is correct |
84 |
Correct |
24 ms |
9560 KB |
Output is correct |
85 |
Correct |
41 ms |
9308 KB |
Output is correct |
86 |
Correct |
42 ms |
9304 KB |
Output is correct |
87 |
Correct |
44 ms |
9444 KB |
Output is correct |
88 |
Correct |
24 ms |
9308 KB |
Output is correct |
89 |
Incorrect |
36 ms |
9560 KB |
Output isn't correct |
90 |
Correct |
37 ms |
9308 KB |
Output is correct |
91 |
Incorrect |
50 ms |
9308 KB |
Output isn't correct |
92 |
Correct |
26 ms |
9308 KB |
Output is correct |
93 |
Correct |
51 ms |
9304 KB |
Output is correct |
94 |
Correct |
39 ms |
9308 KB |
Output is correct |
95 |
Incorrect |
37 ms |
9468 KB |
Output isn't correct |
96 |
Correct |
36 ms |
9308 KB |
Output is correct |
97 |
Correct |
41 ms |
9304 KB |
Output is correct |
98 |
Incorrect |
48 ms |
9308 KB |
Output isn't correct |
99 |
Correct |
43 ms |
9560 KB |
Output is correct |
100 |
Correct |
34 ms |
9304 KB |
Output is correct |
101 |
Correct |
28 ms |
9308 KB |
Output is correct |
102 |
Correct |
48 ms |
9472 KB |
Output is correct |