Codeforces Round #243 DIV1 A問題/DIV2 C問題 - Sereja and Swaps

Source

Codeforces Round #243 DIV1 A問題 (500pt)
Codeforces Round #243 DIV2 C問題 (1500pt)
Problem description

問題概要

$N$ 要素からなる整数列 $\{A_i\}_{i=1}^N$ が与えられる.
最初に $K$ 回以下,任意の $2$ つの要素をスワップして良い.
その後,$\{A_i\}$ の(空でない)連続する部分列の和の最大値を考える.
スワップをうまくやった時,その最大値の最大値を求める問題.

解法

どの位置(開始位置,終了位置)の部分列が最大になるかを $O(N^2)$ 通り全探索する.
位置を決めたら,その部分の和を最大化するようにスワップことを考える.
まず,その部分の中にある要素と外にある要素に分類する.
$K$ 回までスワップできるので,外にある要素は $K$ 個以下使用可能.
よって,外にある要素の中で,大きい方から $K$ 個を中にある要素に加えて,その中から,大きい方から中に最初にあった要素の数だけ取ってきて和を取れば良い.
時間計算量 $O(N^3 \log N)$.

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

#include<cstdio>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<algorithm>
#include<string>
#include<utility>
#include<iostream>
#include<cmath>
#include<ctime>
#include<cassert>
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_nolock()
#define mypc(c) _putchar_nolock(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 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 main(){
  int N, K, A[210];
  int i,j,l,m;
  int res, tmp;
  int in[210], ins, add, out[210], outs;

  reader(&N);
  reader(&K);
  rep(i,N) reader(A+i);

  res = -1000000;
  rep(i,N) REP(j,i,N){
    ins = outs = add = 0;
    rep(m,i) out[outs++] = A[m];
    REP(m,i,j+1) in[ins++] = A[m];
    REP(m,j+1,N) out[outs++] = A[m];
    sort(out, out+outs);
    rep(m,K) if(m < outs) in[ins+add++] = out[outs-1-m];
    sort(in, in+ins+add);
    tmp = 0;
    rep(m,ins) tmp += in[add+m];
    res = max(res, tmp);
  }

  writer(res, '\n');

  return 0;
}

Current time: 2024年03月29日00時27分41秒
Last modified: 2014年05月03日17時18分31秒 (by laycrs)
Tags: Competitive_Programming Codeforces CF243 CF_Div1_A CF_Div2_C
トップページに戻る

Logged in as: unknown user (not login)

ログイン: