Submission #62588

#	Time	Username	Problem	Language	Result	Execution time	Memory
62588		evpipis	곤돌라 (IOI14_gondola)	C++11	55 / 100	28 ms	5756 KiB

gondola

#include "gondola.h"
#include <bits/stdc++.h>
using namespace std;

//#define TEST

#define fi first
#define se second
#define mp make_pair
#define pb push_back
typedef long long ll;
typedef pair<int, int> ii;

vector<ii> vec;

#ifdef TEST
int gondolaSequence[100001];
int replacementSequence[250001];
#endif

int valid(int n, int arr[]){
    int pos = -1;
    for (int i = 0; i < n; i++)
        if (arr[i] <= n) pos = i;

    if (pos != -1){
        int cur = arr[pos];
        for (int i = 0; i < n; i++){
            if (arr[pos] <= n && arr[pos] != cur)
                return 0;

            cur = cur%n+1;
            pos = (pos+1)%n;
        }
    }

    sort(arr, arr+n);
    for (int i = 0; i < n-1; i++)
        if (arr[i] == arr[i+1])
            return 0;
    return 1;
}

//----------------------

int replacement(int n, int arr[], int out[]){
    int pos = -1, nex = n+1, m = 0, cur;
    for (int i = 0; i < n; i++)
        if (arr[i] <= n) pos = i;

    if (pos == -1)
        pos = 0, cur = 1;
    else
        cur = arr[pos];

    for (int i = 0; i < n; i++){
        if (arr[pos] > n)
            vec.pb(mp(arr[pos], cur));

        pos = (pos+1)%n;
        cur = cur%n+1;
    }

    sort(vec.begin(), vec.end());
    for (int i = 0; i < vec.size(); i++)
        while (vec[i].fi > vec[i].se)
            out[m++] = vec[i].se, vec[i].se = nex++;

    return m;
}

//----------------------

int countReplacement(int n, int inputSeq[])
{
  return -3;
}

#ifdef TEST
int main()
{
  int i, n, tag;
  int nr;
  assert(scanf("%d", &tag)==1);
  assert(scanf("%d", &n)==1);
  for(i=0;i< n;i++)
    assert( scanf("%d", &gondolaSequence[i]) ==1);

  switch (tag){
  case 1: case 2: case 3:
    printf("%d\n", valid(n, gondolaSequence));
    break;

  case 4: case 5: case 6:
    nr = replacement(n, gondolaSequence, replacementSequence);
    printf("%d\n", nr);
    if (nr > 0)
      {
	for (i=0; i<nr-1; i++)
	  printf("%d ", replacementSequence[i]);
	printf("%d\n", replacementSequence[nr-1]);
      }
    else printf("\n");
    break;

  case 7: case 8: case 9: case 10:
    printf("%d\n",  countReplacement(n, gondolaSequence));
    break;
  }

  return 0;
}
#endif
/*
3 5
10 4 3 11 12

3 7
2 3 4 9 6 7 1

6 7
2 3 4 9 6 7 1
*/

Compilation message (stderr)

gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:65:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < vec.size(); i++)
                     ~~^~~~~~~~~~~~

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