yukicoder No.110 - しましまピラミッド

Source

ニコニコミュニティ
問題文

問題概要

白い棒が $N$ 本と黒い棒が $M$ 本がある.
白い棒の長さは $W_1,W_2,\ldots,W_N$ で,黒い棒の長さは $B_1,B_2,\ldots,B_M$ である.
棒を重ねてピラミッドを作る.
その際,上に置く棒は下の棒より短い棒でなければいけなく,白黒の棒を交互に使う.
使える棒の数の最大値を求める問題.

解法

白黒,どちらの棒から先に使うかは $2$ パターン両方試す.
下から積み上げていくとして,greedyに使えるものの中から最も長い棒を使っていけば良い.
最初に長さでソートしておけば,前使ったところから調べれば良いので,時間計算量は $O(N \log N + M \log M)$ 程度.

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 mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

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 s=0,m=0;char f[10];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}
template<class T> void writerLn(T x){writer(x,'\n');}

int As, A[20], Bs, B[20];

int main(){
  int i, j, k, loop;
  int res = 0, tmp;
  int a, b;

  reader(&As);
  rep(i,As) reader(A+i);
  reader(&Bs);
  rep(i,Bs) reader(B+i);
  sort(A,A+As);
  sort(B,B+Bs);

  rep(loop,2){
    a = As-1;
    b = Bs-1;
    tmp = 0;
    k = 1000;
    for(;;){
      while(a >= 0 && A[a] >= k) a--;
      if(a < 0) break;
      k = A[a]; tmp++; a--;
      
      while(b >= 0 && B[b] >= k) b--;
      if(b < 0) break;
      k = B[b]; tmp++; b--;
    }
    res = max(res, tmp);
    swap(As, Bs);
    rep(i,max(As,Bs)) swap(A[i],B[i]);
  }

  writerLn(res);

  return 0;
}

Current time: 2017年07月21日13時29分20秒
Last modified: 2014年12月24日00時03分59秒 (by laycrs)
Tags: Competitive_Programming yukicoder
トップページに戻る

Logged in as: unknown user (not login)

ログイン: