Submission #132653

#	Time	Username	Problem	Language	Result	Execution time	Memory
132653		SamAnd	Lottery (CEOI18_lot)	C++17	25 / 100	217 ms	65540 KiB

lot

#include <bits/stdc++.h>
using namespace std;
const int N = 10004;

int n, m;
int a[N];
vector<int> v[N];

int b[N][N];
void ubd(int x1, int x2, int y1, int y2)
{
    if (x1 > x2 || y1 > y2)
        return;
    if (x1 < 0 || x2 < 0 || y1 < 0 || y2 < 0)
        return;
    b[x1][y1]++;
    b[x2 + 1][y2 + 1]--;
}
int p[N][N];

int main()
{
    //freopen("input.txt", "r", stdin);
    scanf("%d%d", &n, &m);
    //n = 2000;
    //m = 1000;
    for (int i = 1; i <= n; ++i)
    {
        //a[i] = rand();
        scanf("%d", &a[i]);
    }
    for (int i = 1; i <= n; ++i)
    {
        for (int j = 1; j <= n; ++j)
        {
            if (a[i] != a[j])
            {
                int x1 = i - m + 1;
                int x2 = i;
                int y1 = j - m + 1;
                int y2 = j;
                int t1 = 0, t2 = 0;
                t1 = max(t1, 1 - x1);
                //t2 = max(t2, x2 - (n - m + 1));
                t1 = max(t1, 1 - y1);
                //t2 = max(t2, y2 - (n - m + 1));
                x1 += t1;
                y1 += t1;
                //x2 -= t2;
                //y2 -= t2;
                ubd(x1, x2, y1, y2);
            }
        }
    }
    for (int i = 1; i <= n - m + 1; ++i)
    {
        for (int j = 1; j <= n - m + 1; ++j)
        {
            p[i][j] = p[i - 1][j - 1] + b[i][j];
            if (i != j)
            {
                v[i].push_back(p[i][j]);
            }
        }
    }
    for (int i = 1; i <= n - m + 1; ++i)
        sort(v[i].begin(), v[i].end());
    int q;
    scanf("%d", &q);
    while (q--)
    {
        int d;
        scanf("%d", &d);
        for (int i = 1; i <= n - m + 1; ++i)
        {
            printf("%d ", upper_bound(v[i].begin(), v[i].end(), d) - v[i].begin());
        }
        printf("\n");
    }
    return 0;
}

Compilation message (stderr)

lot.cpp: In function 'int main()':
lot.cpp:42:29: warning: unused variable 't2' [-Wunused-variable]
                 int t1 = 0, t2 = 0;
                             ^~
lot.cpp:76:82: warning: format '%d' expects argument of type 'int', but argument 2 has type '__gnu_cxx::__normal_iterator<int*, std::vector<int> >::difference_type {aka long int}' [-Wformat=]
             printf("%d ", upper_bound(v[i].begin(), v[i].end(), d) - v[i].begin());
                           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^
lot.cpp:24:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &n, &m);
     ~~~~~^~~~~~~~~~~~~~~~
lot.cpp:30:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &a[i]);
         ~~~~~^~~~~~~~~~~~~
lot.cpp:69:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &q);
     ~~~~~^~~~~~~~~~
lot.cpp:73:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &d);
         ~~~~~^~~~~~~~~~

• Subtask 1: 25 / 25
• Subtask 2: 0 / 20
• Subtask 3: 0 / 20
• Subtask 4: 0 / 15
• Subtask 5: 0 / 20