제출 #636553

#제출 시각아이디문제언어결과실행 시간메모리
636553shmad코알라 (APIO17_koala)C++17
100 / 100
48 ms452 KiB
#include <bits/stdc++.h>

#ifndef ONLINE_JUDGE
	#include "koala.h"
#endif

#define vt vector
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
#define ff first
#define ss second
#define dbg(x) cerr << #x << " = " << x << '\n'
#define bit(x, i) ((x) >> (i) & 1)

using namespace std;
using ll = long long;
using pii = pair<int, int>;
using vvt = vt< vt<int> >;

#ifdef ONLINE_JUDGE

static int N, W;
static int P[105];

static int maxQueries = 3200;
static int numQueries;

static void runGame(int F);
static void grader();

int main() {
    grader();
    return 0;
}

void playRound(int *B, int *R) {
    int i, j;

    int S = 0;
    for (i=0;i<N;++i) {
        if ( !(B[i] >= 0 && B[i] <= W) ) {
            printf("Invalid query.\n");
            exit(0);
        }
        S += B[i];
    }
    if (S > W) {
        printf("Invalid query.\n");
        exit(0);
    }

    numQueries++;
    if (numQueries > maxQueries) {
        printf("Too many queries.\n");
        exit(0);
    }

    int cache[2][205];
    int num[2][205];
    char taken[105][205];

    for (i=0;i<205;++i) {
        cache[1][i] = 0;
        num[1][i] = 0;
    }

    for (i=0;i<N;++i) {
        int v = B[i]+1;
        int ii = i&1;
        int o = ii^1;
        for (j=0;j<=W;++j) {
            cache[ii][j] = cache[o][j];
            num[ii][j] = num[o][j];
            taken[i][j] = 0;
        }
        for (j=W;j>=v;--j) {
            int h = cache[o][j-v] + P[i];
            int hn = num[o][j-v] + 1;
            if (h > cache[ii][j] || (h == cache[ii][j] && hn > num[ii][j])) {
                cache[ii][j] = h;
                num[ii][j] = hn;
                taken[i][j] = 1;
            } else {
                taken[i][j] = 0;
            }
        }
    }

    int cur = W;
    for (i=N-1;i>=0;--i) {
        R[i] = taken[i][cur] ? (B[i] + 1) : 0;
        cur -= R[i];
    }
}

#endif

int a[100], b[100];

int minValue(int n, int w) {  
    fill(a, a + n, 0);
    a[0] = 1;
    playRound(a, b);
    for (int i = 1; i < n; i++) if (b[i] == 0) return i;
    return 0;
}

int maxValue (int n, int w) {
	vt<int> p(n);
	iota(all(p), 0ll);
	while (sz(p) != 1) {
		fill(a, a + n, 0);
		int x = w / sz(p);
		for (int i: p) a[i] = x;
		playRound(a, b);
		p.clear();
		for (int i = 0; i < n; i++) if (b[i] == x + 1) p.pb(i);
	}
    return p[0];
}

int greaterValue (int n, int w) {
	fill(a, a + n, 0);
	int l = 1, r = 13, ans = 0;
	while (l <= r) {
		int mid = l + r >> 1;
		fill(a, a + n, 0);
		a[0] = a[1] = mid;                                               
		playRound(a, b);
		if (b[0] != b[1]) return (b[0] < b[1]);
		if (b[0] > mid && b[1] > mid) ans = mid, l = mid + 1;
		else r = mid - 1;
	}
	a[0] = a[1] = ans;
	playRound(a, b);
	return (b[0] < b[1]);
}

bool comp (int i, int j, int n, int w) {
	fill(a, a + n, 0);
	a[i] = a[j] = w / 2;
	playRound(a, b);
	return (b[j] > w / 2);
}

vt<int> merge_sort (vt<int> &v, int n, int w) {
	if (sz(v) == 1) return v;
	vt<int> l, r;
	for (int i = 0; i < sz(v); i++) {
		if (i < sz(v) / 2) l.pb(v[i]);
		else r.pb(v[i]);
	}
	l = merge_sort(l, n, w);
	r = merge_sort(r, n, w);
	v.clear();
	int i = 0, j = 0;
	while (i < sz(l) && j < sz(r)) v.pb((comp(l[i], r[j], n, w) ? l[i++] : r[j++]));
	while (i < sz(l)) v.pb(l[i++]);
	while (j < sz(r)) v.pb(r[j++]);
	return v;
}

bool check (int x, int l, int r, int w) {
	int mx = r, mn = 1, cnt = r - l + 1;
	while (mx >= l) {
		int sum = 0;
		while (mn < l && sum <= mx && cnt <= x) {
		 	sum += mn;
		 	mn++;
		 	cnt++;
		}
		if (cnt <= x) break;
		cnt -= x + 1;
		mx--;
	}
	return (mx >= l && mx < r);
}

int get (int L, int R, int w) {
	for (int x = 1; x <= w / (R - L + 1); x++) {
		if (check(x, L, R, w)) return x;
	}
	return -1;
}

void solve (vt<int> &v, int l, int r, int n, int w, int *p) {
	if (l == r) {
    	p[v[0]] = l;
    	return;
    }
    int x = get(l, r, w);
    fill(a, a + n, 0);
    for (int i: v) a[i] = x;
    playRound(a, b);
    vt<int> L, R;
    for (int i: v) {
    	if (b[i] == x + 1) R.pb(i);
    	else L.pb(i);
    }
    solve(L, l, l + sz(L) - 1, n, w, p);
    solve(R, r - sz(R) + 1, r, n, w, p);
}

void allValues (int n, int w, int *p) {
    vt<int> v(n);
    iota(all(v), 0);	
    if (w == n * 2) {
        v = merge_sort(v, n, w);
    	for (int i = 0; i < n; i++) p[v[i]] = i + 1;
    	return;
    }
    solve(v, 1, n, n, w, p);
}

#ifdef ONLINE_JUDGE
static void runGame(int F) {
    int i;

    scanf("%d %d",&N,&W);
    for (i=0;i<N;++i) {
        scanf("%d",&P[i]);
    }

    numQueries = 0;
    if (F == 1) {
        printf("%d\n", minValue(N, W));
    } else if (F == 2) {
        printf("%d\n", maxValue(N, W));
    } else if (F == 3) {
        printf("%d\n", greaterValue(N, W));
    } else if (F == 4) {
        int userP[105];
        allValues(N, W, userP);
        for (i=0;i<N;i++) {
            printf("%d ",userP[i]);
        }
        printf("\n");
    }
    printf("Made %d calls to playRound.\n", numQueries);
}

static void grader() {
    int i;

    int F, G;
    scanf("%d %d",&F,&G);

    for (i=0;i<G;i++) {
        runGame(F);
    }
}
#endif

컴파일 시 표준 에러 (stderr) 메시지

koala.cpp: In function 'int greaterValue(int, int)':
koala.cpp:127:15: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  127 |   int mid = l + r >> 1;
      |             ~~^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...