Submission #905978

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
905978	2024-01-13T08:19:50 Z	GrandTiger1729	Teams (CEOI11_tea)	C++17	0 / 100	2500 ms	32368 KB

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

vector<vector<int>> solve(int t, vector<pair<int, int>> a)
{
    int n = a.size();
    vector<vector<int>> ans;
    ans.emplace_back();
    for (int i = 0; i < a[0].first; i++)
    {
        ans.back().push_back(a[i].second);
    }
    int l = a[0].first;
    for (int i = l; i < n; i++)
    {
        if (i - l >= a[l].first)
        {
            ans.emplace_back();
            for (int j = l; j < i; j++)
            {
                ans.back().push_back(a[j].second);
            }
            l = i;
        }
        else if (i + a[i].first < l + a[l].first && ans.back().size() + i - l <= t)
        {
            for (int j = l; j < i; j++)
            {
                ans.back().push_back(a[j].second);
            }
            l = i;
        }
    }
    if (l < n && n - l >= a[l].first)
    {
        ans.emplace_back();
    }
    for (int i = l; i < n; i++)
    {
        ans.back().push_back(a[i].second);
    }
    return ans;
}

int main()
{
    cin.tie(0)->sync_with_stdio(0);
    int n;
    cin >> n;
    vector<pair<int, int>> a(n);
    for (int i = 0; i < n; i++)
    {
        int x;
        cin >> x;
        a[i] = make_pair(x, i);
    }
    sort(a.begin(), a.end(), greater<>());
    int l = a[0].first - 1, r = n;
    int t = solve(n, a).size();
    // while (l < r - 1)
    // {
    //     int mid = (l + r) / 2;
    //     (solve(mid, a).size() == t ? r : l) = mid;
    // }
    // auto ans = solve(r, a);
    int j = n;
    for (int i = l; i < n; i++)
    {
        if (solve(i, a).size() == t)
        {
            j = i;
            break;
        }
    }
    auto ans = solve(j, a);
    cout << ans.size() << '\n';
    for (auto &res : ans)
    {
        cout << res.size() << ' ';
        for (int &i : res)
        {
            cout << i + 1 << ' ';
        }
        cout << '\n';
    }
    return 0;
}

Compilation message

tea.cpp: In function 'std::vector<std::vector<int> > solve(int, std::vector<std::pair<int, int> >)':
tea.cpp:25:79: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   25 |         else if (i + a[i].first < l + a[l].first && ans.back().size() + i - l <= t)
      |                                                     ~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
tea.cpp: In function 'int main()':
tea.cpp:69:32: warning: comparison of integer expressions of different signedness: 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   69 |         if (solve(i, a).size() == t)
      |             ~~~~~~~~~~~~~~~~~~~^~~~
tea.cpp:58:29: warning: unused variable 'r' [-Wunused-variable]
   58 |     int l = a[0].first - 1, r = n;
      |                             ^

Subtask #1
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	452 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Incorrect	1 ms	344 KB	Output isn't correct
4	Halted	0 ms	0 KB	-

Subtask #2
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	1 ms	344 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #3
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	344 KB	Output is correct
2	Incorrect	1 ms	348 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #4
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	21 ms	600 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #5
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	2 ms	604 KB	Output is correct
2	Incorrect	2 ms	604 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #6
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	1738 ms	2916 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #7
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	20 ms	3436 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #8
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Execution timed out	2569 ms	17576 KB	Time limit exceeded
2	Halted	0 ms	0 KB	-

Subtask #9
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	228 ms	30352 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #10
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	241 ms	32368 KB	Output isn't correct
2	Halted	0 ms	0 KB	-