#include <bits/stdc++.h>
#include "dzialka.h"
#include "message.h"
#define FOR(x, b, e) for(int x = b; x <= (e); ++x)
#define FORD(x, b,e) for(int x = b; x >= (e); --x)
#define REP(x, b) for(int x = 0; x < (b); ++x)
#define SIZE(c) ((int)c.size())
#define ALL(c) (c).begin(), (c).end()
#define MP make_pair
#define PB push_back
#define ST first
#define ND second
using namespace std;
typedef long long LL;
typedef vector <int> VI;
typedef pair <int, int> PII;
typedef vector <PII> VPII;
const int MAXN = 5e3 + 9;
bool tab[MAXN][MAXN];
short tab2[MAXN][MAXN];
set <PII> seti;
LL suma = 0;
void clear()
{
suma = 0;
seti.clear();
}
void add(int k)
{
int ile = 0;
while((!seti.empty()) && ((*seti.rbegin()).ST >= k))
{
ile += (*seti.rbegin()).ND;
suma -= (*seti.rbegin()).ST * (*seti.rbegin()).ND;
seti.erase(seti.find((*seti.rbegin())));
}
ile++;
suma += k * ile;
seti.insert(MP(k, ile));
}
int main()
{
if(MyNodeId() != 0)
return 0;
int n = GetFieldHeight(), m = GetFieldWidth();
REP(i, n)
{
REP(j, m)
{
tab[j][i] = IsUsableCell(i, j);
}
}
int pop = 0;
REP(j, m)
{
pop = 0;
FORD(i, n - 1, 0)
{
if(tab[j][i])
pop++;
else
pop = 0;
tab2[j][i] = pop;
}
}
LL wyn = 0;
REP(i, n)
{
clear();
REP(j, m)
{
add(tab2[j][i]);
wyn += suma;
}
}
cout<<wyn<<endl;
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | #include <bits/stdc++.h> #include "dzialka.h" #include "message.h" #define FOR(x, b, e) for(int x = b; x <= (e); ++x) #define FORD(x, b,e) for(int x = b; x >= (e); --x) #define REP(x, b) for(int x = 0; x < (b); ++x) #define SIZE(c) ((int)c.size()) #define ALL(c) (c).begin(), (c).end() #define MP make_pair #define PB push_back #define ST first #define ND second using namespace std; typedef long long LL; typedef vector <int> VI; typedef pair <int, int> PII; typedef vector <PII> VPII; const int MAXN = 5e3 + 9; bool tab[MAXN][MAXN]; short tab2[MAXN][MAXN]; set <PII> seti; LL suma = 0; void clear() { suma = 0; seti.clear(); } void add(int k) { int ile = 0; while((!seti.empty()) && ((*seti.rbegin()).ST >= k)) { ile += (*seti.rbegin()).ND; suma -= (*seti.rbegin()).ST * (*seti.rbegin()).ND; seti.erase(seti.find((*seti.rbegin()))); } ile++; suma += k * ile; seti.insert(MP(k, ile)); } int main() { if(MyNodeId() != 0) return 0; int n = GetFieldHeight(), m = GetFieldWidth(); REP(i, n) { REP(j, m) { tab[j][i] = IsUsableCell(i, j); } } int pop = 0; REP(j, m) { pop = 0; FORD(i, n - 1, 0) { if(tab[j][i]) pop++; else pop = 0; tab2[j][i] = pop; } } LL wyn = 0; REP(i, n) { clear(); REP(j, m) { add(tab2[j][i]); wyn += suma; } } cout<<wyn<<endl; return 0; } |
English