Ruudukko

input/code.cpp: In function 'bool sosanh(item, item)':
input/code.cpp:14:7: warning: suggest explicit braces to avoid ambiguous 'else' [-Wdangling-else]
   14 |     if(a.val == b.val)
      |       ^

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector <ll> vl;
typedef vector <vl> vvl;

struct item
{
    ll val, x, y;
};

bool sosanh(item a, item b)
{
    if(a.val == b.val)
        if(a.x == b.x)
            return a.y < b.y;
        else
            return a.x < b.x;
    return a.val < b.val;
}

void out(vl a)
{
    for(auto x : a)
        cerr << x << " ";
    cerr << "\n";
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    ll n, m, i, j, k = 0, z = 0;
    cin >> n;
    m = n * n;
    vector <item> a(m);
    vvl ans(n, vl(n, 0));
    for(i = 0; i < n; i++)
        for(j = 0; j < n; j++)
        {
            cin >> a[k].val;
            a[k].x = i;
            a[k].y = j;
            k++;
        }
    sort(a.begin(), a.end(), sosanh);
    vvl sx(2, vl(n, 0)), sy(2, vl(n, 0));
    i = j = 0;
    while(i < m)
    {
        while(j < m && a[j].val == a[i].val)
        {
            ans[a[j].x][a[j].y] += sx[0][a[j].x] + sy[0][a[j].y] + 1;
            sx[1][a[j].x] += ans[a[j].x][a[j].y];
            sy[1][a[j].y] += ans[a[j].x][a[j].y];
            j++;
        }
        for(k = 0; k < n; k++)
        {
            sx[0][k] += sx[1][k];
            sx[1][k] = 0;
            sy[0][k] += sy[1][k];
            sy[1][k] = 0;
        }
        //out(sx[0]); out(sy[0]);
        //cerr << i << " " << j << "\n";
        i = j;
    }
    for(auto d : ans)
        for(auto t : d)
            z += t;
    cout << z;
    return 0;
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 2, 3

Verdict: ACCEPTED

Test 5

Group: 2, 3

Verdict: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: ACCEPTED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED