Submission #2699625


Source Code Expand

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

#define NDEBUG
#ifdef DEBUG
#include "../cout11.h"
#undef NDEBUG
#endif
#include <cassert>

typedef long long ll;
typedef long double Double;
typedef unsigned long long ull;
typedef pair<int,int> ii;
typedef pair<ll,ll> llll;
typedef pair<double,double> dd;

typedef vector<int> vi;
typedef vector<vector<int>> vvi;
typedef vector<ii> vii;
typedef vector<vector<ii>> vvii;
typedef vector<ll> vll;
typedef vector<string> vs;
typedef vector<double> vd;
typedef vector<long double> vD;

#define sz(a)  int((a).size())
#define pb  push_back
#define FOR(var,from,to) for(int var=(from);var<=(to);++var)
#define rep(var,n)  for(int var=0;var<(n);++var)
#define rep1(var,n)  for(int var=1;var<=(n);++var)
#define repC2(vari,varj,n)  for(int vari=0;vari<(n)-1;++vari)for(int varj=vari+1;varj<(n);++varj)
#define ALL(c)  (c).begin(),(c).end()
#define RALL(c)  (c).rbegin(),(c).rend()
#define tr(i,c)  for(auto i=(c).begin(); i!=(c).end(); ++i)
#define found(s,e)  ((s).find(e)!=(s).end())
#define mset(arr,val)  memset(arr,val,sizeof(arr))
#define mid(x,y) ((x)+((y)-(x))/2)
#define IN(x,a,b) ((a)<=(x)&&(x)<=(b))

void solve(int H,int W,int D,vvi& a, int Q,vi& l,vi& r) {
    vector<ii> loc(H*W+1);
    rep(r,H) rep(c,W) {
        int x = a[r][c];
        loc[x] = ii(r,c);
    }
    vi d_(H*W+1, 0);
    for (int i=1; i<=H*W-D; ++i) {
        d_[i] = abs(loc[i].first - loc[i+D].first) + abs(loc[i].second - loc[i+D].second);
    }
    vector<ll> da(H*W+1+D, 0);
    for (int b=0; b<D; ++b) {
        for (int i=b; i<=H*W-D; i+=D) {
            da[i+D] = da[i] + d_[i];
        }
    }

    rep(i,Q){ // 1e5
        // int score = 0;
        // for (int x=l[i]; x!=r[i]; x+=D) { // can be 9e4
        //     // x -> x+D
        //     score += d_[x];
        // }
        ll score = da[r[i]] - da[l[i]];
        cout << score << endl;
    }

}

int main() {
    int H,W,D; cin >>H>>W>>D;
    vvi a(H, vi(W));
    rep(h,H) rep(w,W) cin >> a[h][w];
    int Q; cin >>Q;
    vi l(Q), r(Q);
    rep(i,Q) cin >> l[i] >> r[i];
    solve(H,W,D,a, Q,l,r);
    return 0;
}

Submission Info

Submission Time
Task D - Practical Skill Test
User naoya_t
Language C++14 (GCC 5.4.1)
Score 400
Code Size 2182 Byte
Status
Exec Time 241 ms
Memory 4096 KB

Test Cases

Set Name Score / Max Score Test Cases
Sample 0 / 0 sample_01.txt, sample_02.txt, sample_03.txt
All 400 / 400 01.txt, 02.txt, 03.txt, 04.txt, 05.txt, 06.txt, 07.txt, 08.txt, 09.txt, sample_01.txt, sample_02.txt, sample_03.txt
Case Name Status Exec Time Memory
01.txt 241 ms 3712 KB
02.txt 234 ms 3456 KB
03.txt 194 ms 1408 KB
04.txt 229 ms 3456 KB
05.txt 230 ms 4096 KB
06.txt 218 ms 3968 KB
07.txt 225 ms 3968 KB
08.txt 1 ms 256 KB
09.txt 230 ms 3840 KB
sample_01.txt 1 ms 256 KB
sample_02.txt 1 ms 256 KB
sample_03.txt 1 ms 256 KB