Submission #674955

#	Time	Username	Problem	Language	Result	Execution time	Memory
674955		LucaIlie	Chessboard (IZhO18_chessboard)	C++17	70 / 100	668 ms	5784 KiB

chessboard

#include <bits/stdc++.h>

#define int long long

using namespace std;

const int maxK = 1e5;

struct rectangle {
    int l1, c1, l2, c2;
};

int l;
rectangle black[maxK];

int nxt( int x ) {
    if ( x % l == 0 )
        return x;
    return (x / l + 1) * l;
}

int prv( int x ) {
    return x / l * l;
}

long long wbSq( int lin, int col, bool isWhiteCorner ) {
    int x = (lin + 1) / 2, y = col / 2;
    return (x * x + y * y - 2 * x * y) * (isWhiteCorner ? 1 : -1);
}

long long getCost( rectangle r ) {
    int l1 = r.l1, c1 = r.c1, l2 = r.l2, c2 = r.c2;
    long long cost = 0;

    if ( l1 / l == l2 / l && c1 / l == c2 / l )
        return (l2 - l1 + 1) * (c2 - c1 + 1) * ((l1 / l + c1 / l) % 2 == 0 ? -1 : 1);

    if ( l1 % l != 0 && c1 % l != 0 )
        cost += (nxt( l1 ) - l1) * (nxt( c1 ) - c1) * wbSq( 1, 1, (l1 / l + c1 / l) % 2 );
    if ( l1 % l != 0 && c2 % l != 0 )
        cost += (nxt( l1 ) - l1) * (c2 - prv( c2 )) * wbSq( 1, 1, (l1 / l + c2 / l) % 2 );
    if ( l2 % l != 0 && c1 % l != 0 )
        cost += (l2 - prv( l2 )) * (nxt( c1 ) - c1) * wbSq( 1, 1, (l2 / l + c1 / l) % 2 );
    if ( l2 % l != 0 && c2 % l != 0 )
        cost += (l2 - prv( l2 )) * (c2 - prv( c2 )) * wbSq( 1, 1, (l2 / l + c2 / l) % 2 );

    if ( l1 % l != 0 )
        cost += (nxt( l1 ) - l1) * wbSq( 1, prv( c2 ) / l - nxt( c1 ) / l, !(l1 / l + nxt( c1 ) / l) % 2 );
    if ( c1 % l != 0 )
        cost += (nxt( c1 ) - c1) * wbSq( prv( l2 ) / l - nxt( c1 ) / l, 1, !(nxt( l1 ) / l + c1 / l) % 2 );
    if ( l2 % l != 0 )
        cost += (l2 - prv( l2 )) * wbSq( 1, prv( c2 ) / l - nxt( c1 ) / l, !(l2 / l + nxt( c1 ) / l) % 2 );
    if ( c2 % l != 0 )
        cost += (c2 - prv( c2 )) * wbSq( prv( l2 ) / l - nxt( c1 ) / l, 1, !(nxt( l1 ) / l + c2 / l) % 2 );

    cost += l * l * wbSq( prv( l2 ) / l - nxt( c1 ) / l, prv( c2 ) / l - nxt( c1 ) / l, (nxt( l1 ) / l + nxt( l2 ) / l) % 2 );

    printf( "%d %d %d %d %d: %d\n", l, l1, c1, l2, c2, cost );
    return cost;
}

signed main() {
    int n, k;
    long long cost, minCost;

    cin >> n >> k;
    for ( int i = 0; i < k; i++ ) {
        cin >> black[i].l1 >> black[i].c1 >> black[i].l2 >> black[i].c2;
        black[i].l1--;
        black[i].c1--;
        black[i].l2--;
        black[i].c2--;
    }

    minCost = n * n;
    for ( l = 1; l < n; l++ ) {
        if ( n % l == 0 ) {
            int x = (n / l + 1) / 2, y = n / l / 2;

            cost = (l * l) * (x * x + y * y);
            for ( int i = 0; i < k; i++ )
                cost += getCost( black[i] );
            minCost = min( minCost, cost );

            cost = 2 * x * y * l * l;
            for ( int i = 0; i < k; i++ )
                cost -= getCost( black[i] );
            minCost = min( minCost, cost );
        }
    }

    cout << minCost;

    return 0;
}

Compilation message (stderr)

chessboard.cpp: In function 'long long int getCost(rectangle)':
chessboard.cpp:58:15: warning: format '%d' expects argument of type 'int', but argument 2 has type 'long long int' [-Wformat=]
   58 |     printf( "%d %d %d %d %d: %d\n", l, l1, c1, l2, c2, cost );
      |              ~^                     ~
      |               |                     |
      |               int                   long long int
      |              %lld
chessboard.cpp:58:18: warning: format '%d' expects argument of type 'int', but argument 3 has type 'long long int' [-Wformat=]
   58 |     printf( "%d %d %d %d %d: %d\n", l, l1, c1, l2, c2, cost );
      |                 ~^                     ~~
      |                  |                     |
      |                  int                   long long int
      |                 %lld
chessboard.cpp:58:21: warning: format '%d' expects argument of type 'int', but argument 4 has type 'long long int' [-Wformat=]
   58 |     printf( "%d %d %d %d %d: %d\n", l, l1, c1, l2, c2, cost );
      |                    ~^                      ~~
      |                     |                      |
      |                     int                    long long int
      |                    %lld
chessboard.cpp:58:24: warning: format '%d' expects argument of type 'int', but argument 5 has type 'long long int' [-Wformat=]
   58 |     printf( "%d %d %d %d %d: %d\n", l, l1, c1, l2, c2, cost );
      |                       ~^                       ~~
      |                        |                       |
      |                        int                     long long int
      |                       %lld
chessboard.cpp:58:27: warning: format '%d' expects argument of type 'int', but argument 6 has type 'long long int' [-Wformat=]
   58 |     printf( "%d %d %d %d %d: %d\n", l, l1, c1, l2, c2, cost );
      |                          ~^                        ~~
      |                           |                        |
      |                           int                      long long int
      |                          %lld
chessboard.cpp:58:31: warning: format '%d' expects argument of type 'int', but argument 7 has type 'long long int' [-Wformat=]
   58 |     printf( "%d %d %d %d %d: %d\n", l, l1, c1, l2, c2, cost );
      |                              ~^                        ~~~~
      |                               |                        |
      |                               int                      long long int
      |                              %lld

• Subtask 1: 8 / 8
• Subtask 2: 8 / 8
• Subtask 3: 15 / 15
• Subtask 4: 16 / 16
• Subtask 5: 23 / 23
• Subtask 6: 0 / 30