Submission #530137

#	Time	Username	Problem	Language	Result	Execution time	Memory
530137		peuch	Teams (CEOI11_tea)	C++17	80 / 100	578 ms	121660 KiB

tea

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

const int MAXN = 1e6 + 10;

int n;
pair<int, int> v[MAXN];

vector<vector<int> > group;
set<pair<int, int> > tam;

int main(){
  scanf("%d", &n);
  for(int i = 1; i <= n; i++){
    scanf("%d", &v[i].first);
    v[i].second = i;
  }
  sort(v + 1, v + 1 + n);
  reverse(v + 1, v + 1 + n);
  for(int i = 1; i <= n; i++){
    int cur = v[i].second;
    int qnt = v[i].first;
    if(qnt > n - i + 1){
      int grp = tam.begin()->second;
      tam.erase(tam.begin());
      group[grp].push_back(cur);
      tam.insert(make_pair(group[grp].size(), grp));
      continue;
    }
    group.push_back(vector<int> (1, cur));
    int idx = group.size() - 1;
    while(--qnt)
      group[idx].push_back(v[++i].second);
    tam.insert(make_pair(group[idx].size(), idx));
  }
  printf("%lu\n", group.size());
  for(int i = 0; i < group.size(); i++){
    printf("%lu ", group[i].size());
    for(int j = 0; j < group[i].size(); j++)
      printf("%d ", group[i][j]);
    printf("\n");
  }
}

Compilation message (stderr)

tea.cpp: In function 'int main()':
tea.cpp:37:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |   for(int i = 0; i < group.size(); i++){
      |                  ~~^~~~~~~~~~~~~~
tea.cpp:39:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |     for(int j = 0; j < group[i].size(); j++)
      |                    ~~^~~~~~~~~~~~~~~~~
tea.cpp:13:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   13 |   scanf("%d", &n);
      |   ~~~~~^~~~~~~~~~
tea.cpp:15:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   15 |     scanf("%d", &v[i].first);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~

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