AtCoder Beginner Contest 135 (A - Harmony/B - 0 or 1 Swap/C - City Savers)

酒飲みながらでも C 問題まで解けるようになった。

解答

A - Harmony

atcoder.jp

平均ととったやつが答えで、整数にならないやつが "IMPOSSIBLE" になる。

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

int main(){
 
  long long A, B;
  cin >> A >> B;

  if((A + B) % 2 != 0 || (B + A) % 2 != 0){
    cout << "IMPOSSIBLE" << endl;
    return 0;
  }
  
  if(A > B){
    cout << (A + B) / 2 << endl;
  }else{
    cout << (B + A) / 2 << endl;
  }
  return 0;
}

B - 0 or 1 Swap

atcoder.jp
ソートして、下の配列と比べる。異なる部分が 0 個か 2 個なら "YES"。酔っぱらいおじさんだから "Yes"・"No" にしてしまって "WA” 連発した。

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

int main(){
 
  int N, i, a;
  cin >> N;
  vector<int> p(N), _p(N);
  for (i = 0; i < N; i++) {
    cin >> p.at(i);
    _p.at(i) = p.at(i);
  }
  
  sort(_p.begin(), _p.end());
  
  int count = 0;
  for (i = 0; i < N; i++){
    if(_p.at(i) != p.at(i)) count++;
  }
  
  if (count == 2 || count == 0){
    cout << "YES" << endl;
  }else{
    cout << "NO" << endl;
  }
  return 0;
}

C - City Savers

atcoder.jp
1の街のモンスターは1の勇者しか倒せないので、勇者はまず 1 の街に向かう。残った倒せる回数で 2 の街に向かう。次に 2 の勇者が 2 の街に残っているモンスターを倒しにいく。っていうのを繰り返す。

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

int main(){
 
  long long N, i, b1, b2, count = 0;
  cin >> N;
  vector<long long> A(N + 1), B(N);
  for (i = 0; i < N + 1; i++) cin >> A.at(i);
  for (i = 0; i < N; i++) cin >> B.at(i);
  
  for(i = 0; i < N; i++){
    b1 = B.at(i); //勇者が倒せる人数
    B.at(i) = max(0ll, b1 - A.at(i)); //iの街で討伐作業を行った後の勇者の討伐可能数
    count += min(A.at(i), b1);
    A.at(i) = max(A.at(i) - b1, 0ll); //討伐された後の、iの街のモンスターの数
    
    
    b2 = B.at(i);
    count += min(A.at(i + 1), b2);
    A.at(i + 1) = max(A.at(i + 1) - b2, 0ll);
  }
  cout << count << endl;
  return 0;
}

まとめ

できる問題をいくら "AC" しようが成長しねえんだよ、ばーかばーか（反省しています）。