Submission #821078

#	Time	Username	Problem	Language	Result	Execution time	Memory
821078		boris_mihov	Marshmallow Molecules (CCO19_day2problem2)	C++17	5 / 25	4083 ms	177980 KiB

Main

#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <vector>
#include <bitset>
#include <queue>
#include <stack>
#include <set>
#include <map>

typedef long long llong;
const int MAXN = 100000 + 10;
const llong INF = 1e18;
const int INTINF = 1e9;

int n, m;
std::bitset <MAXN> bs[MAXN];

void solve()
{
    llong cnt = 0;
    for (int i = 1 ; i <= n ; ++i)
    {
        std::bitset <MAXN> toOR;
        for (int j = i - 1 ; j >= 1 ; --j)
        {
            if (bs[j][i])
            {
                toOR |= bs[j];
            }
        }
 
        bs[i] |= (toOR >> i + 1) << i + 1;
        cnt += bs[i].count();
        if (((bs[i] >> i + 1) << i + 1).count() == n - i)
        {
            cnt += 1LL * (n - i) * (n - i - 1) / 2;
            break;
        }
    }

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

void input()
{
    std::cin >> n >> m;
    for (int i = 1 ; i <= m ; ++i)
    {
        int u, v;
        std::cin >> u >> v;
        bs[u][v] = 1;
    }
}

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

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

    return 0;
}

Compilation message (stderr)

Main.cpp: In function 'void solve()':
Main.cpp:34:29: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   34 |         bs[i] |= (toOR >> i + 1) << i + 1;
      |                           ~~^~~
Main.cpp:36:26: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   36 |         if (((bs[i] >> i + 1) << i + 1).count() == n - i)
      |                        ~~^~~
Main.cpp:36:49: warning: comparison of integer expressions of different signedness: 'std::size_t' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   36 |         if (((bs[i] >> i + 1) << i + 1).count() == n - i)
      |             ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 0 / 5
• Subtask 3: 0 / 5
• Subtask 4: 0 / 10