AtCoder ABC 173 反省

atcoder.jp

結果

A, B, C, D 問題は 0 WA で AC した。E 問題は解けず。F 問題は見たけど解けそうにないと思ってすぐにあきらめた。
今回で茶色コーダーになりました。

レーティングの推移

感想

茶色になることができて素直にうれしい。緑コーダーを目指して AtCoder 続けていこうと思います。
E 問題は何となく解法は浮かんでいたものの細かい条件分岐等を実装できず駄目だった。率直に言って解けるだけの実力がなかったなと思う。

E 問題復習

問題概要

個の整数が与えられる。その中から 個選んで積をとったときの最大値を求めよ。

解き方

条件分岐が少しややこしい。

1. のとき
2. すべて負の整数で が奇数のとき
3. すべて負の整数で が偶数のとき
4. 上記以外

の 4 パターンを考える必要がある。
1. の場合は簡単で与えられた整数を全て掛ければ終わり。

2. の場合は結果が必ず負になるので、入力を絶対値が小さい順にソートして先頭から 個の積をとればいい。

3. の場合は結果が必ず正になるので、入力を絶対値が大きい順にソートして先頭から 個の積をとればいい。

4. の場合は必ず結果は正になる。
が偶数なら、まず入力を正負で分けて別々の配列に格納し、絶対値の降順でソートする。その後、先頭から2個ずつのペアを作り、正負の両方の配列からペアの積が大きい順に積をとっていけばいい。
が奇数でも、まず入力を正負で分けて別々の配列に格納する。そして正の整数が格納された配列から最大値を 1 つ取り出した後、 が偶数の場合と同様に 個のペアの積をとればいい。最後に、あらかじめ取り出しておいた正の入力の最大値を掛ければ終わり。

ACしたコード

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

// auto mod int
// https://youtu.be/L8grWxBlIZ4?t=9858
// https://youtu.be/ERZuLAxZffQ?t=4807 : optimize
// https://youtu.be/8uowVvQ_-Mo?t=1329 : division
const int mod = 1000000007;
struct mint {
  ll x; // typedef long long ll;
  mint(ll x=0):x((x%mod+mod)%mod){}
  mint operator-() const { return mint(-x);}
  mint& operator+=(const mint a) {
    if ((x += a.x) >= mod) x -= mod;
    return *this;
  }
  mint& operator-=(const mint a) {
    if ((x += mod-a.x) >= mod) x -= mod;
    return *this;
  }
  mint& operator*=(const mint a) { (x *= a.x) %= mod; return *this;}
  mint operator+(const mint a) const { return mint(*this) += a;}
  mint operator-(const mint a) const { return mint(*this) -= a;}
  mint operator*(const mint a) const { return mint(*this) *= a;}
  mint pow(ll t) const {
    if (!t) return 1;
    mint a = pow(t>>1);
    a *= a;
    if (t&1) a *= *this;
    return a;
  }

  // for prime mod
  mint inv() const { return pow(mod-2);}
  mint& operator/=(const mint a) { return *this *= a.inv();}
  mint operator/(const mint a) const { return mint(*this) /= a;}
};
istream& operator>>(istream& is, mint& a) { return is >> a.x;}
ostream& operator<<(ostream& os, const mint& a) { return os << a.x;}

int main(){
  ll n;cin >>n;
  ll k;cin >>k;
  mint ans=1;
  vector<ll>a;
  vector<ll>ap;
  vector<ll>an;
  
  for(ll i=0;i<n;i++){
    ll tmp;cin>>tmp;
    a.push_back(tmp);
    if(tmp<0)an.push_back(tmp);
    else ap.push_back(tmp);
  }
  
  if(n==k){
    for(auto aa:a){
      ans*=aa;
    }
    cout <<ans<<endl;
    return 0;
  }
  
  ll apsize=ap.size();
  if(apsize==0&&k%2==0){
    sort(an.begin(),an.end());
    for(ll i=0;i<k;i++){
      ans*=an.at(i);
    }
    cout <<ans<<endl;
    return 0;
  }
  else if (apsize==0&&k%2==1){
    sort(an.begin(),an.end());
    reverse(an.begin(),an.end());
    for(ll i=0;i<k;i++){
      ans*=an.at(i);
    }
    cout <<ans<<endl;
    return 0;
  }
  
  sort(an.begin(),an.end());
  sort(ap.begin(),ap.end());
  ll apmax;
  if(k%2==1){
    apmax=ap.back();
    ap.pop_back();
  }
  reverse(ap.begin(),ap.end());
  
  ll ansize2=an.size();
  ll apsize2=ap.size();
  
  if(ansize2<=1&&k%2==0){
    for(ll i=0;i<k;i++){
      ans*=ap.at(i);
    }
    cout <<ans<<endl;
    return 0;
  }
  else if(ansize2<=1&&k%2==1){
    for(ll i=0;i<k-1;i++){
      ans*=ap.at(i);
    }
    ans*=apmax;
    cout <<ans<<endl;
    return 0;
  }
  
  vector<ll>aabs;
  for(ll i=0;i+1<ansize2;i+=2){
    aabs.push_back(an.at(i)*an.at(i+1));
  }
  for(ll i=0;i+1<apsize2;i+=2){
    aabs.push_back(ap.at(i)*ap.at(i+1));
  }
  sort(aabs.rbegin(),aabs.rend());
  for(ll i=0;i<k/2;i++){
    ans*=aabs.at(i);
  }
  if(k%2==1)ans*=apmax;
  cout <<ans<<endl;
}

雑記

負の数を割った余りがうまく計算できなかったので有名（？）な mint ライブラリを使ってみた。よく中身を見ておきたい。
解き方の項目があまりに雑な感じになってしまった。もう少し思考の過程を文字に起こせるようになりたい。