Submission #165163

# Submission time Handle Problem Language Result Execution time Memory
165163 2019-11-25T11:59:39 Z Atill83 Bob (COCI14_bob) C++14
120 / 120
179 ms 25096 KB
#include <bits/stdc++.h>
#define ff first
#define ss second
#define endl '\n'
using namespace std;
const long long INF = (long long) 1e18;
const int mod = (int) 1e9+7;
const int MAXN = (int) 1e3+5;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll n;
ll a[MAXN][MAXN];
ll sl[MAXN][MAXN];
ll us[MAXN][MAXN];
ll dig[MAXN][MAXN];
ll ans[MAXN][MAXN];

stack<int> s;
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);

    #ifdef Local
        freopen("../IO/int.txt","r",stdin);
        freopen("../IO/out.txt","w",stdout);
    #endif
    int m;
    cin>>n>>m;

    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= m; j++){
            cin>>a[i][j];
        }
    }
    ll an = 0;
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= m; j++){
            us[i][j] = 1;
            if(i != 0 && a[i - 1][j] == a[i][j]) us[i][j] += us[i - 1][j];
        }
    }
    for(ll i = 1; i <= n; i++){
        for(ll j = 1; j <= m; j++){
            if(j == 0 || a[i][j] != a[i][j - 1]){
                while(!s.empty()) s.pop();
                s.push(j - 1);
                us[i][j - 1] = 0;
                ans[i][j - 1] = 0;
                s.push(j);
                ans[i][j] = us[i][j];
                //cout<<"# ";
            }else{
                while(us[i][s.top()] > us[i][j]){
                    s.pop();
                }
                ll l = s.top();
                //cout<<l<<" ";
                ans[i][j] += (s.empty() ? 0 : ans[i][s.top()]) + (j - l)*us[i][j];
                s.push(j);
            }
            an += ans[i][j];
            //cout<<ans[i][j]<<" ";
        }//cout<<endl;
    }
    cout<<an<<endl;
    #ifdef Local
        cout<<endl<<fixed<<setprecision(2)<<1000.0 * clock() / CLOCKS_PER_SEC<< " milliseconds ";
    #endif
}
# Verdict Execution time Memory Grader output
1 Correct 3 ms 1016 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 3 ms 1016 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 32 ms 12152 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 35 ms 12296 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 39 ms 12152 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 39 ms 12084 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 153 ms 23904 KB Output is correct
2 Correct 98 ms 24792 KB Output is correct
3 Correct 98 ms 25096 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 179 ms 23928 KB Output is correct
2 Correct 98 ms 24184 KB Output is correct
3 Correct 93 ms 24184 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 173 ms 23928 KB Output is correct
2 Correct 95 ms 24172 KB Output is correct
3 Correct 94 ms 24200 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 178 ms 23936 KB Output is correct
2 Correct 98 ms 24224 KB Output is correct
3 Correct 95 ms 25020 KB Output is correct