Submission #731563

#	Time	Username	Problem	Language	Result	Execution time	Memory
731563		danikoynov	Aliens (IOI16_aliens)	C++14	4 / 100	2 ms	2008 KiB

aliens

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

typedef long long ll;

const int maxn = 1e5 + 10;

struct interval
{
    ll left, right;

    interval(ll _left = 0, ll _right = 0)
    {
        left = _left;
        right = _right;
    }
};

bool cmp(const interval &it1, const interval &it2)
{
    if (it1.left == it2.left)
        return it1.right > it2.right;
    return it1.left < it2.left;
}

const ll inf = 1e18;

interval h[maxn];
ll dp[1010][1010];

ll area(ll left, ll right)
{
    if (left > right)
        return 0;
    return (right - left + 1) * (right - left + 1);
}
ll take_photos(int n, int m, int k, vector<int> r, vector<int> c)
{
    vector < interval > vec;
    for (int i = 0; i < n; i ++)
    {
        if (r[i] > c[i])
            swap(r[i], c[i]);
        interval cur(r[i], c[i]);
        vec.push_back(cur);
    }

    sort(vec.begin(), vec.end(), cmp);
    n = 0;
    int to = -1;
    for (int i = 0; i < vec.size(); i ++)
    {
        if (vec[i].right > to)
        {
            to = vec[i].right;
            h[++ n] = vec[i];
        }
    }

    ///for (int i = 1; i <= n; i ++)
       /// cout << h[i].left << " : " << h[i].right << endl;

    for (int i = 0; i <= n; i ++)
        for (int j = 0; j <= k; j ++)
        dp[i][j] = inf;

    k = min(k, n);
    dp[0][0] = 0;
    h[0] = interval(-1, -1);
    for (int i = 1; i <= n; i ++)
    {
        for (int j = 1; j <= k; j ++)
        {
            for (int p = 1; p <= i; p ++)
            {
                dp[i][j] = min(dp[i][j], dp[p - 1][j - 1] + area(h[p].left, h[p].right) - area(h[p].left, h[p - 1].right));
            }
        }
    }

    /**for (int i = 1; i <= n; i ++, cout << endl)
        for (int j = 1; j <= k; j ++)
        cout << dp[i][j] << " ";*/


    return dp[n][k];
}

Compilation message (stderr)

aliens.cpp: In function 'll take_photos(int, int, int, std::vector<int>, std::vector<int>)':
aliens.cpp:52:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<interval>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |     for (int i = 0; i < vec.size(); i ++)
      |                     ~~^~~~~~~~~~~~

• Subtask 1: 4 / 4
• Subtask 2: 0 / 12
• Subtask 3: 0 / 9
• Subtask 4: 0 / 16
• Subtask 5: 0 / 19
• Subtask 6: 0 / 40