Submission #962250

# Submission time Handle Problem Language Result Execution time Memory
962250 2024-04-13T09:42:37 Z danikoynov Art Class (IOI13_artclass) C++14
76 / 100
49 ms 11444 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(1018.01, 1022.23, 1191.85);
    avg_dev[2] = pixel(424.553, 383.047, 273.954);
    avg_dev[3] = pixel(2216.33, 2146.01, 1979.27);
    avg_dev[4] = pixel(67.4286, 55.6262, 91.9571);
}

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; 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);
          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;
}
# Verdict Execution time Memory Grader output
1 Incorrect 37 ms 10956 KB Output isn't correct
2 Incorrect 37 ms 10928 KB Output isn't correct
3 Correct 33 ms 10568 KB Output is correct
4 Correct 27 ms 10288 KB Output is correct
5 Incorrect 47 ms 11220 KB Output isn't correct
6 Correct 35 ms 10684 KB Output is correct
7 Correct 26 ms 10324 KB Output is correct
8 Correct 33 ms 10832 KB Output is correct
9 Incorrect 41 ms 11000 KB Output isn't correct
10 Correct 31 ms 10700 KB Output is correct
11 Correct 36 ms 10836 KB Output is correct
12 Correct 29 ms 10708 KB Output is correct
13 Correct 33 ms 10580 KB Output is correct
14 Correct 31 ms 10836 KB Output is correct
15 Incorrect 42 ms 11444 KB Output isn't correct
16 Incorrect 32 ms 10576 KB Output isn't correct
17 Correct 35 ms 10916 KB Output is correct
18 Correct 46 ms 11240 KB Output is correct
19 Correct 34 ms 10764 KB Output is correct
20 Incorrect 35 ms 10848 KB Output isn't correct
21 Incorrect 41 ms 11092 KB Output isn't correct
22 Correct 43 ms 11148 KB Output is correct
23 Correct 38 ms 10836 KB Output is correct
24 Correct 31 ms 10520 KB Output is correct
25 Correct 34 ms 10832 KB Output is correct
26 Correct 23 ms 10224 KB Output is correct
27 Correct 26 ms 10500 KB Output is correct
28 Correct 32 ms 10836 KB Output is correct
29 Incorrect 49 ms 11204 KB Output isn't correct
30 Correct 35 ms 11044 KB Output is correct
31 Incorrect 27 ms 10580 KB Output isn't correct
32 Correct 35 ms 10684 KB Output is correct
33 Correct 32 ms 10444 KB Output is correct
34 Correct 32 ms 10684 KB Output is correct
35 Correct 46 ms 11392 KB Output is correct
36 Correct 26 ms 10328 KB Output is correct
37 Correct 36 ms 10724 KB Output is correct
38 Correct 36 ms 11092 KB Output is correct
39 Correct 10 ms 9816 KB Output is correct
40 Incorrect 41 ms 11088 KB Output isn't correct
41 Correct 35 ms 10896 KB Output is correct
42 Correct 32 ms 10560 KB Output is correct
43 Correct 37 ms 11120 KB Output is correct
44 Incorrect 42 ms 11096 KB Output isn't correct
45 Correct 43 ms 10828 KB Output is correct
46 Correct 36 ms 11092 KB Output is correct
47 Correct 23 ms 10320 KB Output is correct
48 Incorrect 34 ms 10984 KB Output isn't correct
49 Correct 29 ms 10364 KB Output is correct
50 Incorrect 41 ms 11088 KB Output isn't correct
51 Incorrect 34 ms 10952 KB Output isn't correct
52 Correct 29 ms 10832 KB Output is correct
53 Correct 35 ms 11220 KB Output is correct
54 Correct 35 ms 10916 KB Output is correct
55 Correct 27 ms 10464 KB Output is correct
56 Correct 36 ms 10836 KB Output is correct
57 Correct 34 ms 10844 KB Output is correct
58 Incorrect 36 ms 11064 KB Output isn't correct
59 Correct 39 ms 11036 KB Output is correct
60 Correct 32 ms 10832 KB Output is correct
61 Correct 35 ms 10836 KB Output is correct
62 Incorrect 39 ms 11084 KB Output isn't correct
63 Correct 28 ms 10496 KB Output is correct
64 Correct 36 ms 10840 KB Output is correct
65 Correct 35 ms 10648 KB Output is correct
66 Correct 30 ms 10512 KB Output is correct
67 Correct 35 ms 11032 KB Output is correct
68 Correct 35 ms 10788 KB Output is correct
69 Correct 33 ms 10840 KB Output is correct
70 Correct 34 ms 10832 KB Output is correct
71 Incorrect 27 ms 10280 KB Output isn't correct
72 Correct 27 ms 10324 KB Output is correct
73 Correct 44 ms 11180 KB Output is correct
74 Correct 24 ms 10324 KB Output is correct
75 Incorrect 43 ms 11016 KB Output isn't correct
76 Correct 23 ms 10328 KB Output is correct
77 Correct 20 ms 10500 KB Output is correct
78 Correct 28 ms 10332 KB Output is correct
79 Incorrect 41 ms 11216 KB Output isn't correct
80 Correct 35 ms 11092 KB Output is correct
81 Correct 34 ms 10836 KB Output is correct
82 Correct 35 ms 11152 KB Output is correct
83 Correct 11 ms 9816 KB Output is correct
84 Correct 23 ms 10284 KB Output is correct
85 Correct 24 ms 10288 KB Output is correct
86 Correct 37 ms 10976 KB Output is correct
87 Correct 32 ms 10836 KB Output is correct
88 Correct 35 ms 11276 KB Output is correct
89 Correct 36 ms 11096 KB Output is correct
90 Correct 34 ms 10580 KB Output is correct
91 Incorrect 43 ms 11344 KB Output isn't correct
92 Correct 18 ms 10076 KB Output is correct
93 Correct 28 ms 10608 KB Output is correct
94 Correct 33 ms 10644 KB Output is correct
95 Correct 36 ms 11096 KB Output is correct
96 Correct 28 ms 10500 KB Output is correct
97 Correct 33 ms 10836 KB Output is correct
98 Correct 34 ms 11092 KB Output is correct
99 Correct 33 ms 10672 KB Output is correct
100 Correct 43 ms 11056 KB Output is correct
101 Correct 42 ms 11088 KB Output is correct
102 Correct 22 ms 10332 KB Output is correct