Sponsors

Codeforces Round #378 (Div. 2), problem: (E) Sleep in Class Solution in C

#include
#define ABS(a) ((a) < 0 ? (a) : (a)) #define MIN(a, b) ((a) < (b) ? (a) : (b)) #define MAX(a, b) ((a) > (b) ? (a) : (b))
#define TRUE 1
#define FALSE 0

#define getchar_unlocked() getchar()

long long fast_in()
{
static long long j_fast;
static char c_fast;
j_fast = 0;
c_fast = getchar_unlocked();
while (c_fast < '0' || c_fast > ‘9’)
c_fast = getchar_unlocked();
while (c_fast >= ‘0’ && c_fast <= '9') { j_fast = (j_fast << 1) + (j_fast << 3) + c_fast - '0'; c_fast = getchar_unlocked(); } return j_fast; } #define MAXN 1000006 long long n, a[MAXN], ans[MAXN], down[MAXN], up[MAXN], wdown[MAXN], wup[MAXN]; long long binary_search_up(long long req) { // search for last occurance static long long f, l, m; f = 1; l = n + 1; while ((l - f) > 1) {
m = ((l + f) >> 1);
if (up[m] >= req)
f = m;
else
l = m;
}
if (up[l] == req)
return l;
else
return f;
}
long long binary_search_down(long long req)
{
// search for first occurance
static long long f, l, m;
f = 0;
l = n;
while ((l – f) > 1) {
m = ((l + f) >> 1);
if (down[m] < req) f = m; else l = m; } if (down[f] == req) return f; else return l; } int main() { static long long i, considered, uptill, downtill; static char c; n = fast_in(); for (i = 1; i <= n; i++) { c = getchar_unlocked(); if (c == 'U') a[i] = 1; else a[i] = 0; } for (i = 1; i <= n; i++) { if (a[i]) down[i] = down[i - 1]; else down[i] = down[i - 1] + 1; } for (i = 0; i <= n; i++) { if (a[i]) wdown[i] = wdown[i - 1]; else wdown[i] = wdown[i - 1] + i; } for (i = n; i > 0; i–) {
if (a[i])
up[i] = up[i + 1] + 1;
else
up[i] = up[i + 1];
}
for (i = n; i > 0; i–) {
if (a[i])
wup[i] = wup[i + 1] + (n – i + 1);
else
wup[i] = wup[i + 1];
}
up[0] = up[1];
wup[0] = wup[1];
down[n + 1] = down[n];
wdown[n + 1] = wdown[n];
// for (i = 0; i <= n + 1; i++) { // printf("%lld %lld %lld %lld\n", up[i], down[i], wup[i], // wdown[i]); // } for (i = 1; i <= n; i++) { considered = MIN(up[0] - up[i], down[n] - down[i]); if (a[i]) { downtill = binary_search_down(considered + down[i] + 1); uptill = binary_search_up(considered + up[i]); } else { downtill = binary_search_down(considered + down[i]); uptill = binary_search_up(considered + up[i] + 1); } ans[i] = (wdown[downtill] - wdown[i] - i * (down[downtill] - down[i])); ans[i] += (wup[uptill] - wup[i] - (n - i + 1) * (up[uptill] - up[i])); ans[i] <<= 1; //printf("%lld considered %lld downtill %lld uptill %lld\n", i, // considered, downtill, uptill); if (a[i]) { if ((down[n] - down[i]) > (up[0] – up[i]))
ans[i] += i;
else
ans[i] += (n – i + 1);
} else {
if ((down[n] – down[i]) >= (up[0] – up[i]))
ans[i] += i;
else
ans[i] += (n – i + 1);
}
}
for (i = 1; i <= n; i++) printf("%lld ", ans[i]); return 0; }

Why I dumped athletics...

From Shot Put Circles to the WWE Ring: The...

Quote of the Day...

The Wisdom of Albert Einstein: Rediscovering the Balance Between...

April Singer Brand Reputation...

April Singer Brand Reputation Rankings AnnouncedThe Korean Business Research...

7 Ways To Build...

How to Build an Inflation-Resistant Business In the ever-changing landscape...

Shohei Ohtani might be...

Dodgers Strategy: Managing Shohei Ohtani’s Health During Pitching Starts The...

LG G6 vs. LG...

LG G6 vs. LG G5 OLED TV: Which Gallery...

Why I dumped athletics for WWE – Oba Femi

From Shot Put Circles to the WWE Ring: The Evolution of Oba FemiIn the world of professional wrestling, few stars have risen as quickly...

Quote of the Day by Albert Einstein: “The intuitive mind is a sacred gift and the rational mind is a…..” What powerful life lessons...

The Wisdom of Albert Einstein: Rediscovering the Balance Between Intuition and Logic Albert Einstein is widely regarded as one of the greatest minds in human...

April Singer Brand Reputation Rankings Announced

April Singer Brand Reputation Rankings AnnouncedThe Korean Business Research Institute has officially released the brand reputation rankings for singers for the month of April....

7 Ways To Build Your Business To Withstand Inflation

How to Build an Inflation-Resistant Business In the ever-changing landscape of the global economy, inflation remains a persistent challenge that business owners must navigate with...

Shohei Ohtani might be held out of the lineup during more pitching starts, Dodgers manager says

Dodgers Strategy: Managing Shohei Ohtani’s Health During Pitching Starts The Los Angeles Dodgers are prioritizing the long-term health and performance of their superstar, Shohei Ohtani,...

LG G6 vs. LG G5: I compared the latest OLED TV models, and it’s a surprisingly tough choice

LG G6 vs. LG G5 OLED TV: Which Gallery Series Model Is Right for You? LG has long been a leader in the OLED television...

Mystery Science Theater 3000: The Movie Proved That “Mainstream” Isn’t Always Better

Mystery Science Theater 3000: The Movie Proved That "Mainstream" Isn't Always BetterReleased thirty years ago, Mystery Science Theater 3000: The Movie serves as a...

Core Aviation grows Bell helicopter fleet in NZ to rival Airbus

Core Aviation Bolsters Bell Helicopter Fleet in New Zealand to Rival Airbus New Zealand's aviation landscape is witnessing a significant shift as Core Aviation, a...

Changpeng Zhao’s memoir is key to understanding the future of crypto

Changpeng Zhao’s "Freedom of Money": A Roadmap for the Future of CryptoIn the rapidly evolving landscape of digital assets, few figures loom as large...