Submission #1007288

#	Time	Username	Problem	Language	Result	Execution time	Memory
1007288		VMaksimoski008	Nafta (COI15_nafta)	C++17	34 / 100	1026 ms	39940 KiB

nafta

#include <bits/stdc++.h>

#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
//#define int long long

using namespace std;

using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using Comp = array<ll, 3>;

const int mod = 1e9 + 7;
const int LOG = 20;
const int maxn = 1e5 + 5;

int dr[8] = {0, -1, 0, 1, 1, 1, -1, -1};
int dc[8] = {-1, 0, 1, 0, 1, -1, 1, -1};
char mat[2005][2005];
bool vis[2005][2005];
vector<Comp> comp;
int L=1e9, R=-1, sum=0, n, m;

void dfs(int r, int c) {
    vis[r][c] = 1;
    L = min(L, c); R = max(R, c);
    sum += (mat[r][c] - '0');

    for(int i=0; i<4; i++) {
        int newR = r + dr[i];
        int newC = c + dc[i];
        if(newR < 0 || newR >= n || newC < 0 || newC >= m) continue;
        if(vis[newR][newC] || mat[newR][newC] == '.') continue;
        dfs(newR, newC);
    }
}

struct BIT {
    int n;
    vector<int> tree;

    void config(int _n) {
        n = _n + 10;
        tree.resize(_n+60);
    }

    void update(int p, int v) {
        for(p++; p<n; p+=p&-p) tree[p] += v;
    }

    int query(int p) {
        int ans = 0;
        for(p++; p>0; p-=p&-p) ans += tree[p];
        return ans;
    }

    int query(int l, int r) { return query(r) - query(l-1); }
};


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

    cin >> n >> m;
    for(int i=0; i<n; i++)
        for(int j=0; j<m; j++) cin >> mat[i][j];

    for(int i=0; i<n; i++) {
        for(int j=0; j<m; j++) {
            if(vis[i][j] || mat[i][j] == '.') continue;
            L=1e9, R=-1, sum=0;
            dfs(i, j);
            comp.push_back({ L + 1, R + 1, sum });
        }
    }

    int dp[m+1][m+1];
    memset(dp, 0, sizeof(dp));
    sort(comp.begin(), comp.end(), [&](Comp &a, Comp &b) { return a[1] < b[1]; });
    int ptr = 0;

    BIT bit; bit.config(m);
    for(auto &[l, r, v] : comp) bit.update(l, v);

    for(int i=1; i<=m; i++) {
        while(ptr < comp.size() && comp[ptr][1] < i) {
            bit.update(comp[ptr][0], -comp[ptr][2]);
            ptr++;
        }
        
        for(int j=1; j<=i; j++) {
            for(int k=0; k<i; k++) {
                dp[i][j] = max(dp[i][j], dp[k][j-1] + bit.query(k+1, i));
            }
        }
    }
    
    for(int i=1; i<=m; i++) {
        int mx = 0;
        for(int j=1; j<=m; j++) mx = max(mx, dp[j][i]);
        cout << mx << '\n';
    }
    return 0;
}

Compilation message (stderr)

nafta.cpp: In function 'int main()':
nafta.cpp:88:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::array<long long int, 3> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   88 |         while(ptr < comp.size() && comp[ptr][1] < i) {
      |               ~~~~^~~~~~~~~~~~~

• Subtask 1: 11 / 11
• Subtask 2: 23 / 23
• Subtask 3: 0 / 66