Submission #151610

#	Time	Username	Problem	Language	Result	Execution time	Memory
151610		evpipis	Nuclearia (CEOI15_nuclearia)	C++98	92 / 100	1131 ms	874968 KiB

nuclearia

#include <bits/stdc++.h>
using namespace std;

#define fi first
#define se second
#define mp make_pair
#define pb push_back
typedef pair<int, int> ii;
typedef long long ll;
typedef pair<ll, ll> pll;

const int len = 2e5+5;
pair<ii, ii> arr[len];
vector<vector<ll> > pref;
int n, m, k;

struct node{
    ll ax, ay, b;

    node(ll ax_ = 0, ll ay_ = 0, ll b_ = 0){
        ax = ax_;
        ay = ay_;
        b = b_;
    }

    void operator+=(const node &other){
        ax += other.ax;
        ay += other.ay;
        b += other.b;
    }

    void operator-=(const node &other){
        ax -= other.ax;
        ay -= other.ay;
        b -= other.b;
    }
};

vector<vector<node> > ver, diag1, diag2;

int main(){
    scanf("%d %d %d", &m, &n, &k);
    for (int i = 0; i < k; i++)
        scanf("%d %d %d %d", &arr[i].fi.se, &arr[i].fi.fi, &arr[i].se.fi, &arr[i].se.se);

    pref.resize(n+1);
    for (int i = 0; i <= n; i++)
        pref[i].resize(m+1, 0);

    /*hor.resize(n+2);
    for (int i = 0; i <= n+1; i++)
        hor[i].resize(m+2, node());
    */

    ver.resize(n+2);
    for (int i = 0; i <= n+1; i++)
        ver[i].resize(m+2, node());

    diag1.resize(n+2);
    for (int i = 0; i <= n+1; i++)
        diag1[i].resize(m+2, node());

    diag2.resize(n+2);
    for (int i = 0; i <= n+1; i++)
        diag2[i].resize(m+2, node());

    for (int i = 0; i < k; i++){
        int r = arr[i].fi.fi, c = arr[i].fi.se;
        int a = arr[i].se.fi, b = arr[i].se.se;
        int s = a/b;
        int r0 = max(1, r-s), r1 = min(n, r+s);
        int c0 = max(1, c-s), c1 = min(m, c+s);

        // case 1
        node val = node(0, b, a-c*1LL*b);
        ver[r0][c0] += val;
        ver[r1+1][c0] -= val;

        // case 4
        val = node(0, b, -a-b*1LL*c);
        if (c1 < m){
            ver[r0][c1+1] += val;
            ver[r1+1][c1+1] -= val;
        }

        // case 2
        val = node(b, -b, b*1LL*c-b*1LL*r);
        if ((r-r0) <= (c-c0)){
            diag1[r0][c-r+r0] += val;
        }
        else{
            diag1[r-c+c0][c0] += val;
            ver[r0][c0] += val;
            ver[r-c+c0][c0] -= val;
        }

        if ((r1-r) <= (c1-c)){
            diag1[r1+1][c+r1-r+1] -= val;
        }

        // case 3
        val = node(-b, -b, b*1LL*c+b*1LL*r);
        if ((r1-r) <= (c-c0)){
            diag2[r1][c-r1+r] += val;
        }
        else{
            diag2[r+c-c0][c0] += val;
            ver[r+c-c0+1][c0] += val;
            ver[r1+1][c0] -= val;
        }

        if ((r-r0) <= (c1-c)){
            diag2[r0-1][c+r-r0+1] -= val;
        }
    }

    for (int x = 1; x <= n; x++)
        for (int y = 1; y <= m; y++){
            ver[x][y] += ver[x-1][y];

            //hor[x][y] = hor[x][y] + ver[x][y];
        }

    for (int y = 1; y <= m; y++)
        for (int x = 1; x <= n; x++){
            diag1[x][y] += diag1[x-1][y-1];
            diag2[x][y] += diag2[x+1][y-1];

            ver[x][y] += diag1[x][y];
            ver[x][y] += diag2[x][y];
        }

    for (int x = 1; x <= n; x++)
        for (int y = 1; y <= m; y++){
            ver[x][y] += ver[x][y-1];

            pref[x][y] = ver[x][y].ax*x + ver[x][y].ay*y + ver[x][y].b;
            pref[x][y] += pref[x-1][y]+pref[x][y-1]-pref[x-1][y-1];
        }

    int q;
    scanf("%d", &q);
    for (int i = 0; i < q; i++){
        int x1, y1, x2, y2;
        scanf("%d %d %d %d", &y1, &x1, &y2, &x2);
        ll sum = (pref[x2][y2]-pref[x1-1][y2]-pref[x2][y1-1]+pref[x1-1][y1-1]);
        ll many = (x2-x1+1)*1LL*(y2-y1+1);
        ll ans = sum/many;
        if ((sum%many)*2 >= many)
            ans++;
        printf("%lld\n", ans);
    }
    return 0;
}

/*
4 5 1
3 2 9 1
*/

Compilation message (stderr)

nuclearia.cpp: In function 'int main()':
nuclearia.cpp:42:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d %d %d", &m, &n, &k);
     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
nuclearia.cpp:44:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d %d %d %d", &arr[i].fi.se, &arr[i].fi.fi, &arr[i].se.fi, &arr[i].se.se);
         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
nuclearia.cpp:142:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &q);
     ~~~~~^~~~~~~~~~
nuclearia.cpp:145:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d %d %d %d", &y1, &x1, &y2, &x2);
         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 3 / 3
• Subtask 2: 2 / 2
• Subtask 3: 3 / 3
• Subtask 4: 2 / 2
• Subtask 5: 6 / 6
• Subtask 6: 4 / 4
• Subtask 7: 6 / 6
• Subtask 8: 4 / 4
• Subtask 9: 15 / 15
• Subtask 10: 10 / 10
• Subtask 11: 15 / 15
• Subtask 12: 10 / 10
• Subtask 13: 12 / 12
• Subtask 14: 0 / 8