寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #banner - 横断幕 (Banner)

JOI 2010-2011 日本代表選手選考会 (春合宿代替措置) 過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#banner - 横断幕 (Banner)

https://www.ioi-jp.org/camp/2011/2011-sp-tasks/2011-sp-day1.pdf

僕が作成、提出したコードは、以下のとおりです。

/*
AtCoder Problems in C++
#banner - 横断幕 (Banner)
https://www.ioi-jp.org/camp/2011/2011-sp-tasks/2011-sp-day1.pdf
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
int main(void){
  ll h,w;
  cin>>h>>w;
  vector<vector<ll>> banner(h,vector<ll>(w));
  for(ll i=0;i<h;i++){
    for(ll j=0;j<w;j++){
      cin>>banner[i][j];
    }
  }
  ll ans=0;
  for(ll j=0;j<w;j++){
    for(ll k=j+1;k<w;k++){
      vector<ll> v(8);
      for(ll i=0;i<h;i++){
        ll f=(1<<banner[i][j])|(1<<banner[i][k]);
        for(ll l=0;l<8;l++){
          if((l|f)==7){
            ans+=v[l];
          }
        }
        v[f]++;
      }
    }
  }
  cout<<ans<<endl;
  return 0;
}