Submission #969251

#	Time	Username	Problem	Language	Result	Execution time	Memory
969251		anango	Art Class (IOI13_artclass)	C++17	100 / 100	62 ms	9816 KiB

artclass

#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(420.707, 429.94, 475.923);
    avg_dev[2] = pixel(255.938, 241.82, 189.427);
    avg_dev[3] = pixel(1563.29, 1532.6, 1435.09);
    avg_dev[4] = pixel(25.4556, 23.6549, 35.4133);
}
 
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 = 4;
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;
}
 
pixel get_average_pixel()
{
     pixel s;
     for (int i = 0; i < h; i ++)
          for (int j = 0; j < w; j ++)
     {
          s.r += image[i][j].r;
          s.g += image[i][j].g;
          s.b += image[i][j].b;
     }
 
     double cnt = (double)(h * w);
     s.r /= cnt;
     s.g /= cnt;
     s.b /= cnt;
     return s;
}
 
 
int further_check()
{
     pixel avg = get_average_pixel();
 
     penalty[1] = penalty[2] = 0;
     for (int i = 1; i <= 2; i ++)
     {
          double dr = abs(avg.r - avg_st[i].r);
          double dg = abs(avg.g - avg_st[i].g);
          double db = abs(avg.b - avg_st[i].b);
 
          dr = dr * dr;
          dg = dg * dg;
          db = db * db;
          penalty[i] = dr + dg + db;
     }
 
     if (penalty[1] < penalty[2])
          return 1;
     return 2;
}
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;
    }
 
    if (mx == 1 || mx == 2)
     return further_check();
    return mx;
}

• Subtask 1: 100 / 100