Submission #155790

#	Time	Username	Problem	Language	Result	Execution time	Memory
155790		georgerapeanu	JOIRIS (JOI16_joiris)	C++11	30 / 100	3 ms	504 KiB

joiris

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;

int n,k;
int a[55];
int tmp[55];
int cnt[55];
int ord[55];
int growth[55];

vector<pair<int,int> > op;

int dist(int a,int b){
    if(a > b){
        b += k;
    }
    return b - a;
}

void get_same_height_range(){
    int ma = 0;
    for(int i = 1;i <= n;i++){
        ma = max(ma,a[i]);
    }

    for(int i = 1;i <= n;i++){
        while(ma - a[i] >= k){
            a[i] += k;
            op.push_back({1,i});
        }
    }
}

void paralel(vector<pair<int,int> > &v){
    memset(growth,0,sizeof(growth));
    for(auto it:v){
        for(int x = 1;x <= it.second;x++){
            op.push_back({2,it.first});
        }
        for(int j = it.first;j < it.first + k;j++){
            a[j] += it.second;
            growth[j] += it.second;
        }
    }

    sort(ord + 1,ord + 1 + n,[&](int x,int y){return growth[x] < growth[y];});

    for(int i = 1;i <= n;i++){
        if(growth[ord[i]] > 0){
            continue;
        }
        a[ord[i]] += k;
        op.push_back({1,ord[i]});
    }
}

void do_tetris(){
    int mi = a[1];
    for(int i = 1;i <= n;i++){
        mi = min(mi,a[i]);
    }
    for(int i = 1;i <= n;i++){
        a[i] -= mi;
    }
}

int main(){

    scanf("%d %d",&n,&k);

    for(int i = 1;i <= n;i++){
        scanf("%d",&a[i]);
    }

    int real_h = -1;

    for(int h = 0;h < k;h++){
        for(int i = 1;i <= n;i++){
            tmp[i] = a[i] % k;
        }
        for(int i = 1;i <= n - k + 1;i++){
            cnt[i] = dist(tmp[i],h);
            for(int j = i;j < i + k;j++){
                tmp[j] += cnt[i];
                tmp[j] %= k;
            }
        }
        bool ok = true;
        for(int i = 1;i <= n;i++){
            ord[i] = i;
            if(tmp[i] != h){
                ok = false;
                break;
            }
        }
        if(ok){
            real_h = h;
            break;
        }
    }

    if(real_h == -1){
        printf("-1\n");
        return 0;
    }

    for(int i = 1;i <= n;i++){
        get_same_height_range();
        do_tetris();
        vector<pair<int,int> > pos;
        for(int j = i;j <= i;j += k){
            pos.push_back({j,cnt[j]});
        }
        paralel(pos);
        do_tetris();
    }

    get_same_height_range();

    printf("%d\n",(int)op.size());
    for(auto it:op){
        printf("%d %d\n",it.first,it.second);
    }

    return 0;
}

Compilation message (stderr)

joiris.cpp: In function 'int main()':
joiris.cpp:73:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d %d",&n,&k);
     ~~~~~^~~~~~~~~~~~~~~
joiris.cpp:76:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d",&a[i]);
         ~~~~~^~~~~~~~~~~~

• Subtask 1: 15 / 15
• Subtask 2: 15 / 15
• Subtask 3: 0 / 15
• Subtask 4: 0 / 55