oj.uz

Submission #921757

# Submission time Handle Problem Language Result Execution time Memory
921757 2024-02-04T09:53:26 Z boris_mihov Naan (JOI19_naan) C++17
0 / 100
 2 ms 860 KB 
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <vector>

typedef long long llong;
const int MAXN = 2000 + 10;
const int INF = 1e9;

int n, l;
llong maxSum;
struct Fraction
{
    llong p;    
    llong q;
    Fraction()
    {
        p = 0;
        q = 1;
    }    

    Fraction(llong _p, llong _q = 1)
    {
        p = _p;
        q = _q;
        llong g = std::__gcd(p, q);
        p /= g; q /= g;

        llong x = maxSum * p;
        x = x / q + ((x % q) > 0);
        p = x;
        q = maxSum;
    }

    friend Fraction operator + (const Fraction &left, const Fraction &right)
    {
        return {left.p * right.q + right.p * left.q, left.q * right.q};
    }

    friend Fraction operator - (const Fraction &left, const Fraction &right)
    {
        return {left.p * right.q - right.p * left.q, left.q * right.q};
    }

    friend Fraction operator * (const Fraction &left, const Fraction &right)
    {
        return {left.p * right.p, left.q * right.q};
    }

    friend Fraction operator / (const Fraction &left, const Fraction &right)
    {
        return {left.p * right.q, left.q * right.p};
    }

    friend void operator += (Fraction &left, const Fraction &right)
    {
        left = left + right;
    }

    friend void operator -= (Fraction &left, const Fraction &right)
    {
        left = left - right;
    }

    friend void operator *= (Fraction &left, const Fraction &right)
    {
        left = left * right;
    }
    
    friend void operator /= (Fraction &left, const Fraction &right)
    {
        left = left / right;
    }

    friend bool operator < (const Fraction &left, const Fraction &right)
    {
        return (left.p * right.q < right.p * left.q);
    }

    friend bool operator <= (const Fraction &left, const Fraction &right)
    {
        return (left.p * right.q <= right.p * left.q);
    }

    void print()
    {
        std::cout << p << ' ' << q << '\n';
    }
};

bool taken[MAXN];
Fraction s[MAXN];
llong v[MAXN][MAXN];
Fraction nextSum[MAXN];
Fraction currentSum[MAXN];
std::vector <Fraction> positions;
std::vector <int> order;

void solve()
{
    for (int i = 1 ; i <= l ; ++i)
    {
        llong sum = 0;
        for (int j = 1 ; j <= n ; ++j)
        {
            sum += v[j][i];
        }

        maxSum = std::max(maxSum, sum);
    }

     for (int i = 1 ; i <= n ; ++i)
    {
        for (int j = 1 ; j <= l ; ++j)
        {
            s[i] += v[i][j];
        }

        s[i] /= n;
    }

    Fraction takenTo = 0;
    for (int pos = 1 ; pos <= l ; ++pos)
    {
        Fraction minimum = l + 1;
        int idx = -1;

        for (int i = 1 ; i <= n ; ++i)
        {
            if (taken[i])
            {
                continue;
            }

            nextSum[i] = currentSum[i] + (1 - takenTo) * v[i][pos];
            if (s[i] <= nextSum[i])
            {
                Fraction must = pos - 1 + takenTo + (s[i] - currentSum[i]) / v[i][pos];

                if (must < minimum)
                {
                    minimum = must;
                    idx = i;
                }
            }
        }

        if (idx == -1)
        {
            takenTo = 0;
            for (int i = 1 ; i <= n ; ++i)
            {
                currentSum[i] = nextSum[i];
            }
        } else
        {
            taken[idx] = true;
            order.push_back(idx);
            positions.push_back(minimum);
            takenTo = minimum - pos + 1;
            std::fill(currentSum + 1, currentSum + n + 1, 0);
            pos--;
        }
    }

    positions.pop_back();
}

void print()
{
    for (int i = 0 ; i < n - 1 ; ++i)
    {
        assert(std::__gcd(positions[i].p, positions[i].q) == 1);
        positions[i].print();
    }

    for (int i = 0 ; i < n ; ++i)
    {
        std::cout << order[i] << ' ';
    }

    std::cout << '\n';
}

void input()
{
    std::cin >> n >> l;
    for (int i = 1 ; i <= n ; ++i)
    {
        for (int j = 1 ; j <= l ; ++j)
        {
            std::cin >> v[i][j];
        }
    }
}

void fastIOI()
{
    std::ios_base :: sync_with_stdio(0);
    std::cout.tie(nullptr);
    std::cin.tie(nullptr);
}

int main()
{
    fastIOI();
    input();
    solve();
    print();

    return 0;
}

Subtask #1
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Runtime error 2 ms 856 KB Execution killed with signal 6
3 Halted 0 ms 0 KB -

Subtask #2
0 / 24.0

# Verdict Execution time Memory Grader output
1 Runtime error 2 ms 860 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #3
0 / 71.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Runtime error 2 ms 856 KB Execution killed with signal 6
3 Halted 0 ms 0 KB -