Codeforces Round #244 DIV2 B問題 - Prison Transfer

Source

Codeforces Round #244 DIV2 B問題 (1000pt)
Problem description

問題概要

$N$ 人の囚人が $1$ 列に並んでいる.
$k$ 番目の囚人の罪の重さ $A_k$ が与えられる.
連続する $C$ 人の囚人で,全員の罪の重さが $T$ 以下であるような選び方は何通りあるかを求める問題.

解法

各場所について,その場所から左 $C$ 人のうち,何人の罪の重さが $T$ 以下かを計算していく.
左から $C$ 番目の人については愚直に計算して,それ以降は,場所が一つ右に動くことで,追加される人,除外される人を見て,差分を用いて計算する.
時間計算量は $O(N)$.

C++のスパゲッティなコード

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

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define READER_BUF_SIZE 1048576
#define WRITER_BUF_SIZE 1048576
int reader_pt=READER_BUF_SIZE,reader_last;char reader_buf[READER_BUF_SIZE];
int writer_pt=0;char writer_buf[WRITER_BUF_SIZE];
#define mygc(c) {if(reader_pt==READER_BUF_SIZE)reader_pt=0,reader_last=fread(reader_buf,sizeof(char),READER_BUF_SIZE,stdin);(c)=reader_buf[reader_pt++];}
#define mypc(c) {if(writer_pt==WRITER_BUF_SIZE)writer_pt=0,fwrite(writer_buf,sizeof(char),WRITER_BUF_SIZE,stdout);writer_buf[writer_pt++]=(c);}
#define myed {fwrite(writer_buf,sizeof(char),writer_pt,stdout);writer_pt=0;}

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void writer(int x, char c){int i,sz=0,m=0;char buf[10];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}

int in[300000];

int main(){
  int N, T, C;
  int i, k, res;

  reader(&N);
  reader(&T);
  reader(&C);
  rep(i,N) reader(in+i);
  rep(i,N) in[i] = (in[i]<=T?1:0);
  
  k = res = 0;
  rep(i,C) k += in[i];
  if(k==C) res++;
  REP(i,C,N){
    k += in[i] - in[i-C];
    if(k==C) res++;
  }
  writer(res, '\n');

  myed;
  return 0;
}

Current time: 2024年03月29日22時38分10秒
Last modified: 2014年05月03日18時57分14秒 (by laycrs)
Tags: Competitive_Programming Codeforces CF244 CF_Div2_B
トップページに戻る

Logged in as: unknown user (not login)

ログイン: