oj.uz

Submission #962280

# Submission time Handle Problem Language Result Execution time Memory
962280 2024-04-13T10:00:58 Z danikoynov Art Class (IOI13_artclass) C++14
78 / 100
 91 ms 11356 KB 
#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(1006.35, 1021.6, 1187.03);
    avg_dev[2] = pixel(428.742, 386.04, 275.814);
    avg_dev[3] = pixel(2262.22, 2191.14, 2022.99);
    avg_dev[4] = pixel(57.7061, 52.7395, 85.8614);
}

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 = 8;
pixel get_average_block_deviation()
{
     pixel avg;
     double cnt = 0.0;
         for (int i = 0; i < h - block + 1; i ++)
     for (int j = 0; j < w - block + 1; j ++)
     {
          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);
          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();
    for (int s = 1; s <= 4; s ++)
    {
         double dr = abs(dev.r - avg_dev[s].r);
         double dg = abs(dev.g - avg_dev[s].g);
         double db = abs(dev.b - avg_dev[s].b);

         dr = dr * dr;
         dg = dg * dg;
         db = db * db;
         penalty[s] += dr + dg + db;
    }
    int mx = 1;
    for (int i = 1; i <= 4; i ++)
    {
         if (penalty[i] < penalty[mx])
          mx = i;
    }
    return mx;
}

Subtask #1
78.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 58 ms 10580 KB Output is correct
2 Correct 16 ms 9816 KB Output is correct
3 Correct 67 ms 10932 KB Output is correct
4 Correct 58 ms 10584 KB Output is correct
5 Correct 32 ms 10072 KB Output is correct
6 Correct 60 ms 10832 KB Output is correct
7 Correct 62 ms 10832 KB Output is correct
8 Correct 77 ms 10976 KB Output is correct
9 Correct 33 ms 10196 KB Output is correct
10 Correct 86 ms 10992 KB Output is correct
11 Incorrect 45 ms 10628 KB Output isn't correct
12 Incorrect 69 ms 10996 KB Output isn't correct
13 Correct 70 ms 10992 KB Output is correct
14 Correct 48 ms 10664 KB Output is correct
15 Correct 62 ms 10836 KB Output is correct
16 Correct 41 ms 10328 KB Output is correct
17 Incorrect 62 ms 10980 KB Output isn't correct
18 Correct 67 ms 11088 KB Output is correct
19 Correct 80 ms 10960 KB Output is correct
20 Correct 62 ms 10832 KB Output is correct
21 Incorrect 73 ms 11092 KB Output isn't correct
22 Correct 37 ms 10396 KB Output is correct
23 Correct 62 ms 11216 KB Output is correct
24 Correct 67 ms 10804 KB Output is correct
25 Correct 58 ms 10792 KB Output is correct
26 Incorrect 71 ms 10988 KB Output isn't correct
27 Correct 74 ms 11100 KB Output is correct
28 Correct 50 ms 10736 KB Output is correct
29 Correct 74 ms 11088 KB Output is correct
30 Correct 64 ms 10832 KB Output is correct
31 Correct 46 ms 10732 KB Output is correct
32 Correct 64 ms 10980 KB Output is correct
33 Incorrect 74 ms 11092 KB Output isn't correct
34 Correct 48 ms 10588 KB Output is correct
35 Incorrect 62 ms 10828 KB Output isn't correct
36 Correct 62 ms 10832 KB Output is correct
37 Incorrect 76 ms 10996 KB Output isn't correct
38 Incorrect 71 ms 11004 KB Output isn't correct
39 Correct 60 ms 10992 KB Output is correct
40 Incorrect 84 ms 11356 KB Output isn't correct
41 Incorrect 74 ms 10832 KB Output isn't correct
42 Correct 31 ms 10076 KB Output is correct
43 Correct 65 ms 10992 KB Output is correct
44 Correct 57 ms 10680 KB Output is correct
45 Correct 60 ms 10736 KB Output is correct
46 Correct 61 ms 10832 KB Output is correct
47 Correct 61 ms 10996 KB Output is correct
48 Correct 66 ms 10840 KB Output is correct
49 Correct 46 ms 10584 KB Output is correct
50 Incorrect 62 ms 10984 KB Output isn't correct
51 Correct 62 ms 10784 KB Output is correct
52 Correct 64 ms 10992 KB Output is correct
53 Correct 54 ms 10584 KB Output is correct
54 Correct 73 ms 10792 KB Output is correct
55 Correct 78 ms 11216 KB Output is correct
56 Correct 53 ms 10992 KB Output is correct
57 Incorrect 65 ms 10956 KB Output isn't correct
58 Correct 57 ms 10952 KB Output is correct
59 Correct 43 ms 10496 KB Output is correct
60 Correct 63 ms 10876 KB Output is correct
61 Correct 56 ms 10992 KB Output is correct
62 Incorrect 91 ms 11064 KB Output isn't correct
63 Correct 46 ms 10584 KB Output is correct
64 Correct 39 ms 10332 KB Output is correct
65 Correct 67 ms 10972 KB Output is correct
66 Correct 68 ms 10980 KB Output is correct
67 Incorrect 48 ms 10548 KB Output isn't correct
68 Incorrect 78 ms 11088 KB Output isn't correct
69 Correct 64 ms 10988 KB Output is correct
70 Incorrect 76 ms 10992 KB Output isn't correct
71 Correct 72 ms 10992 KB Output is correct
72 Correct 65 ms 10860 KB Output is correct
73 Correct 62 ms 10844 KB Output is correct
74 Incorrect 76 ms 11224 KB Output isn't correct
75 Incorrect 57 ms 10736 KB Output isn't correct
76 Correct 55 ms 10720 KB Output is correct
77 Correct 45 ms 10568 KB Output is correct
78 Correct 60 ms 11004 KB Output is correct
79 Correct 50 ms 10580 KB Output is correct
80 Correct 60 ms 10924 KB Output is correct
81 Correct 76 ms 11348 KB Output is correct
82 Correct 62 ms 10732 KB Output is correct
83 Correct 49 ms 10588 KB Output is correct
84 Correct 70 ms 10988 KB Output is correct
85 Correct 59 ms 10988 KB Output is correct
86 Correct 70 ms 10988 KB Output is correct
87 Incorrect 86 ms 11148 KB Output isn't correct
88 Correct 43 ms 10544 KB Output is correct
89 Correct 50 ms 10996 KB Output is correct
90 Correct 45 ms 10528 KB Output is correct
91 Correct 64 ms 10988 KB Output is correct
92 Correct 56 ms 10836 KB Output is correct
93 Correct 57 ms 10992 KB Output is correct
94 Correct 60 ms 10868 KB Output is correct
95 Correct 39 ms 10328 KB Output is correct
96 Correct 18 ms 9820 KB Output is correct
97 Correct 61 ms 10992 KB Output is correct
98 Correct 55 ms 10832 KB Output is correct
99 Correct 65 ms 10836 KB Output is correct
100 Correct 68 ms 10940 KB Output is correct
101 Correct 67 ms 10836 KB Output is correct
102 Correct 52 ms 10580 KB Output is correct