Submission #877902

#	Time	Username	Problem	Language	Result	Execution time	Memory
877902		konber	T-Covering (eJOI19_covering)	C++14	30 / 100	68 ms	8524 KiB

covering

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>

using namespace std;

vector<vector<int>> a, grid;
vector<pair<int, int>> sp;
int K, N, M;
int memo[5003][5];

int dp(int i, int prev){
    if(i==K) return 0;
    if(memo[i][prev] != -2) return memo[i][prev];
    int s1=-1, s2=-1, s3=-1, s4=-1;
    int x=sp[i].first, y=sp[i].second;
    int k=sp[i-1].first, j=sp[i-1].second;
    if(prev==1){
        grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=1;
        if(x-1>=0 && x+1<M && y-1>=0 && grid[x-1][y]+grid[x+1][y]+grid[x][y-1]==0){
            grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=0;
            int p=dp(i+1, 1);
            if(p != -1)
                s1 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y-1]+p;
        }
        grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=1;
        if(x-1>=0 && x+1<M && y+1<N && grid[x-1][y]+grid[x+1][y]+grid[x][y+1]==0){
            grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=0;
            int p=dp(i+1, 2);
            if(p!=-1)
                s2 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y+1]+p;
        }
        grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=1;
        if(y-1>=0 && y+1<N && x-1>=0 && grid[x][y-1]+grid[x][y+1]+grid[x-1][y]==0){
            grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=0;
            int p=dp(i+1, 3);
            if(p!=-1)
                s3 = a[x][y]+a[x][y-1]+a[x][y+1]+a[x-1][y]+p;
        }
        grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=1;
        if(y-1>=0 && y+1<N && x+1<M && grid[x][y-1]+grid[x][y+1]+grid[x+1][y]==0){
            grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=0;
            int p=dp(i+1, 4);
            if(p!=-1)
                s4 = p+a[x][y]+a[x][y-1]+a[x][y+1]+a[x+1][y];
        }
        grid[k][j]=grid[k-1][j]=grid[k+1][j]=grid[k][j-1]=0;
    }
    if(prev==2){
        grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=1;
        if(x-1>=0 && x+1<M && y-1>=0 && grid[x-1][y]+grid[x+1][y]+grid[x][y-1]==0){
            grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=0;
            int p=dp(i+1, 1);
            if(p != -1)
                s1 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y-1]+p;
        }
        grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=1;
        if(x-1>=0 && x+1<M && y+1<N && grid[x-1][y]+grid[x+1][y]+grid[x][y+1]==0){
            grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=0;
            int p=dp(i+1, 2);
            if(p!=-1)
                s2 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y+1]+p;
        }
        grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=1;
        if(y-1>=0 && y+1<N && x-1>=0 && grid[x][y-1]+grid[x][y+1]+grid[x-1][y]==0){
            grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=0;
            int p=dp(i+1, 3);
            if(p!=-1)
                s3 = a[x][y]+a[x][y-1]+a[x][y+1]+a[x-1][y]+p;
        }
        grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=1;
        if(y-1>=0 && y+1<N && x+1<M && grid[x][y-1]+grid[x][y+1]+grid[x+1][y]==0){
            grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=0;
            int p=dp(i+1, 4);
            if(p!=-1)
                s4 = p+a[x][y]+a[x][y-1]+a[x][y+1]+a[x+1][y];
        }
        grid[k][j] = grid[k-1][j]=grid[k+1][j]=grid[k][j+1]=0;
    }
    if(prev==3){
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=1;
        if(x-1>=0 && x+1<M && y-1>=0 && grid[x-1][y]+grid[x+1][y]+grid[x][y-1]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=0;
            int p=dp(i+1, 1);
            if(p != -1)
                s1 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y-1]+p;
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=1;
        if(x-1>=0 && x+1<M && y+1<N && grid[x-1][y]+grid[x+1][y]+grid[x][y+1]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=0;
            int p=dp(i+1, 2);
            if(p!=-1)
                s2 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y+1]+p;
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=1;
        if(y-1>=0 && y+1<N && x-1>=0 && grid[x][y-1]+grid[x][y+1]+grid[x-1][y]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=0;
            int p=dp(i+1, 3);
            if(p!=-1)
                s3 = a[x][y]+a[x][y-1]+a[x][y+1]+a[x-1][y]+p;
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=1;
        if(y-1>=0 && y+1<N && x+1<M && grid[x][y-1]+grid[x][y+1]+grid[x+1][y]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=0;
            int p=dp(i+1, 4);
            if(p!=-1)
                s4 = p+a[x][y]+a[x][y-1]+a[x][y+1]+a[x+1][y];
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k-1][j]=0;
    }
    if(prev==4){
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=1;
        if(x-1>=0 && x+1<M && y-1>=0 && grid[x-1][y]+grid[x+1][y]+grid[x][y-1]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=0;
            int p=dp(i+1, 1);
            if(p != -1)
                s1 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y-1]+p;
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=1;
        if(x-1>=0 && x+1<M && y+1<N && grid[x-1][y]+grid[x+1][y]+grid[x][y+1]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=0;
            int p=dp(i+1, 2);
            if(p!=-1)
                s2 = a[x][y]+a[x-1][y]+a[x+1][y]+a[x][y+1]+p;
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=1;
        if(y-1>=0 && y+1<N && x-1>=0 && grid[x][y-1]+grid[x][y+1]+grid[x-1][y]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=0;
            int p=dp(i+1, 3);
            if(p!=-1)
                s3 = a[x][y]+a[x][y-1]+a[x][y+1]+a[x-1][y]+p;
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=1;
        if(y-1>=0 && y+1<N && x+1<M && grid[x][y-1]+grid[x][y+1]+grid[x+1][y]==0){
            grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=0;
            int p=dp(i+1, 4);
            if(p!=-1)
                s4 = p+a[x][y]+a[x][y-1]+a[x][y+1]+a[x+1][y];
        }
        grid[k][j]=grid[k][j-1]=grid[k][j+1]=grid[k+1][j]=0;
    }
    int d=max(s1, max(s2, max(s3, s4)));
    return memo[i][prev] = d;
}


int f(int x){
    if(x==K) return 0;
    int i=sp[x].first, j=sp[x].second;
    int s1=-1, s2=-1, s3=-1, s4=-1;
    if(i-1>=0 && i+1<M && j-1>=0){
        if(grid[i][j]+grid[i-1][j]+grid[i+1][j]+grid[i][j-1]==0){
            grid[i][j]=grid[i-1][j]=grid[i+1][j]=grid[i][j-1]=1;
            int d=f(x+1);
            if(d!=-1) s1 = a[i][j]+a[i-1][j]+a[i+1][j]+a[i][j-1]+d;
            grid[i][j]=grid[i-1][j]=grid[i+1][j]=grid[i][j-1]=0;
        }
    }
    if(i-1>=0 && i+1<M && j+1<N){
        if(grid[i][j]+grid[i-1][j]+grid[i+1][j]+grid[i][j+1]==0){
            grid[i][j]=grid[i-1][j]=grid[i+1][j]=grid[i][j+1]=1;
            int d=f(x+1);
            if(d!=-1) s2 = a[i][j]+a[i-1][j]+a[i+1][j]+a[i][j+1]+d;
            grid[i][j]=grid[i-1][j]=grid[i+1][j]=grid[i][j+1]=0;
        }
    }
    if(j-1>=0 && j+1<N && i+1<M){
        if(grid[i][j]+grid[i][j-1]+grid[i][j+1]+grid[i+1][j]==0){
            grid[i][j]=grid[i][j-1]=grid[i][j+1]=grid[i+1][j]=1;
            int d=f(x+1);
            if(d!=-1) s3 = a[i][j]+a[i][j-1]+a[i][j+1]+a[i+1][j]+d;
            grid[i][j]=grid[i][j-1]=grid[i][j+1]=grid[i+1][j]=0;
        }
    }
    if(j-1>=0 && j+1<N && i-1>=0){
        if(grid[i][j]+grid[i][j-1]+grid[i][j+1]+grid[i-1][j]==0){
            grid[i][j]=grid[i][j-1]=grid[i][j+1]=grid[i-1][j]=1;
            int d=f(x+1);
            if(d!=-1) s4 = a[i][j]+a[i][j-1]+a[i][j+1]+a[i-1][j]+d;
            grid[i][j]=grid[i][j-1]=grid[i][j+1]=grid[i-1][j]=0;
        }
    }
    return max(s1, max(s2, max(s3, s4)));
}

int main()
{
    scanf("%d%d", &M, &N);
    a.resize(M, vector<int>(N));
    grid.resize(M, vector<int>(N));
    for(int i=0; i < M; i++){
        for(int j=0; j < N; j++)
            scanf("%d", &a[i][j]);
    }
    scanf("%d", &K);
    sp.resize(K);
    for(int i=0; i < K; i++){
        scanf("%d%d", &sp[i].first, &sp[i].second);
    }
    sort(sp.begin(), sp.end());
    if(K <= 10){
        int ans=f(0);
        if(ans>=0) printf("%d\n", ans);
        else printf("No\n");
    }
    else if(sp[0].first==sp.back().first){
        memset(memo, -2, sizeof memo);
        int s1=-1, s2=-1, s3=-1, s4=-1;
        int i=sp[0].first, j=sp[0].second;
        if(i-1>=0 && i+1<M && j-1>=0) s1=a[i][j]+a[i-1][j]+a[i+1][j]+a[i][j-1]+dp(1, 1);
        if(i-1>=0 && i+1<M && j+1<N) s2=a[i][j]+a[i-1][j]+a[i+1][j]+a[i][j+1]+dp(1, 2);
        if(j-1>=0 && j+1<N && i-1>=0) s3=a[i][j]+a[i][j-1]+a[i][j+1]+a[i-1][j]+dp(1, 3);
        if(j-1>=0 && j+1<N && i+1>M) s4=a[i][j]+a[i][j-1]+a[i][j+1]+a[i+1][j]+dp(1, 4);
        int ans=max(s1, max(s2, max(s3, s4)));
        if(ans==-1) printf("No\n");
        else printf("%d\n", ans);
    }
    else{
        for(int i=0; i < K; i++){
            grid[sp[i].first][sp[i].second] = 1;
        }
        int sum=0;
        bool valid=true;
        for(int x=0; x < K; x++){
            int i=sp[x].first, j=sp[x].second;
            int s1=-1, s2=-1, s3=-1, s4=-1;
            if(i-1>=0 && i+1<M && j-1>=0 && grid[i-1][j]+grid[i+1][j]+grid[i][j-1]==0)
                s1=a[i][j]+a[i-1][j]+a[i+1][j]+a[i][j-1];
            if(i-1>=0 && i+1<M && j+1<N && grid[i-1][j]+grid[i+1][j]+grid[i][j+1]==0)
                s2=a[i][j]+a[i-1][j]+a[i+1][j]+a[i][j+1];
            if(j-1>=0 && j+1<N && i-1>=0 && grid[i][j-1]+grid[i][j+1]+grid[i-1][j]==0)
                s3=a[i][j]+a[i][j-1]+a[i][j+1]+a[i-1][j];
            if(j-1>=0 && j+1<N && i+1<M && grid[i][j-1]+grid[i][j+1]+grid[i+1][j]==0)
                s4=a[i][j]+a[i][j-1]+a[i][j+1]+a[i+1][j];
            int d=max(s1, max(s2, max(s3, s4)));
            if(d==-1){
                valid=false;
                break;
            }
            sum += d;
        }
        if(!valid) printf("No\n");
        else printf("%d\n", sum);
    }
}

Compilation message (stderr)

covering.cpp: In function 'int main()':
covering.cpp:189:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  189 |     scanf("%d%d", &M, &N);
      |     ~~~~~^~~~~~~~~~~~~~~~
covering.cpp:194:18: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  194 |             scanf("%d", &a[i][j]);
      |             ~~~~~^~~~~~~~~~~~~~~~
covering.cpp:196:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  196 |     scanf("%d", &K);
      |     ~~~~~^~~~~~~~~~
covering.cpp:199:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  199 |         scanf("%d%d", &sp[i].first, &sp[i].second);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 0 / 10
• Subtask 4: 0 / 10
• Subtask 5: 15 / 15
• Subtask 6: 0 / 20
• Subtask 7: 0 / 30