結果

問題 No.662 スロットマシーン
コンテスト
ユーザー algon_320
提出日時 2018-03-09 23:16:46
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,793 bytes
コンパイル時間 2,321 ms
コンパイル使用メモリ 195,304 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-10 19:11:22
合計ジャッジ時間 9,271 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 2 WA * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define stoi stoll
using ll=long long;
using vi=vector<int>;
using pii=pair<int,int>;
#define ALL(c) begin(c),end(c)
#define RALL(c) rbegin(c),rend(c)
#define ITR(i,b,e) for(auto i=(b);i!=(e);++i)
#define FORE(x,c) for(auto &x:c)
#define REPF(i,a,n) for(int i=a,i##len=(int)(n);i<i##len;++i)
#define REP(i,n) REPF(i,0,n)
#define REPR(i,n) for(int i=(int)(n);i>=0;--i)
#define SZ(c) ((int)c.size())
#define CONTAIN(c,x) (c.find(x)!=end(c))
#define OUTOFRANGE(y,x,h,w) ((y)<0||(x)<0||(y)>=(h)||(x)>=(w))
#define dump(...)
const signed INF_=1001001001; const ll INF=1001001001001001001LL;
const int DX[9]={0,1,0,-1,1,1,-1,-1,0},DY[9]={-1,0,1,0,-1,1,1,-1,0};
template<class T> ostream& operator<<(ostream &os,const vector<T> &v) {
    ITR(i,begin(v),end(v))os<<*i<<(i==end(v)-1?"":" ");return os;}
template<class T> istream& operator>>(istream &is,vector<T> &v) {
    ITR(i,begin(v),end(v)) is>>*i;return is;}
template<class T,class U> istream& operator>>(istream &is, pair<T,U> &p) {
    is>>p.first>>p.second;return is;}
template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}return 0;}
template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}return 0;}
template<class T> using heap=priority_queue<T,vector<T>,greater<T>>;
struct{template<class T> operator T(){T x;cin>>x;return x;}} IN;
struct before_main_function {
    before_main_function() {
        cin.tie(0);ios::sync_with_stdio(false);
        cout<<setprecision(15)<<fixed;
        #define endl "\n"
    }
} before_main_function;
//------------------8<------------------------------------8<--------------------

signed main() {
    map<string,int> id;
    vector<int> coin(5);
    REP(i,5) {
        string s;
        int c;
        cin>>s>>c;
        id[s]=i;
        coin[i]=c;
    }
    int na,nb,nc;
    vector<int> A,B,C;
    map<vector<int>,int> cnt_a,cnt_b,cnt_c;
    cin>>na;
    REP(i,na) {
        string s;
        cin>>s;
        A.push_back(id[s]);
    }
    REP(i,na) {
        vector<int> tmp;
        REP(j,3) tmp.push_back(A[(i+j)%na]);
        cnt_a[tmp]++;
    }
    cin>>nb;
    REP(i,nb) {
        string s;
        cin>>s;
        B.push_back(id[s]);
    }
    REP(i,nb) {
        vector<int> tmp;
        REP(j,3) tmp.push_back(B[(i+j)%nb]);
        cnt_b[tmp]++;
    }
    cin>>nc;
    REP(i,nc) {
        string s;
        cin>>s;
        C.push_back(id[s]);
    }
    REP(i,nc) {
        vector<int> tmp;
        REP(j,3) tmp.push_back(C[(i+j)%nc]);
        cnt_c[tmp]++;
    }

    double ans=0;
    vector<int> task2(5,0);
    vector<vector<int>> b(3);
    function<void(int)> dfs=[&](int i) {
        if(i==3) {
            if(!CONTAIN(cnt_a,b[0]) || !CONTAIN(cnt_b,b[1]) || !CONTAIN(cnt_c,b[2])) {
                return;
            }

            double score=0;
            int num=cnt_a[b[0]]*cnt_b[b[1]]*cnt_c[b[2]];
            double p=num;
            p/=na*nb*nc;

            REP(j,3) {
                if(b[0][j]==b[1][j] && b[1][j]==b[2][j]) {
                    task2[b[0][j]]+=num;
                    score+=coin[b[0][j]]*num;
                }
            }
            if(b[0][0]==b[1][1] && b[1][1]==b[2][2]) {
                task2[b[0][0]]+=num;
                score+=coin[b[0][0]]*num;
            }
            if(b[0][2]==b[1][1] && b[1][1]==b[2][0]) {
                task2[b[0][2]]+=num;
                score+=coin[b[0][2]]*num;
            }

            ans+=p*score;
            return;
        }
        REP(j,5) {
            REP(k,5) {
                REP(l,5) {
                    b[i]={j,k,l};
                    dfs(i+1);
                }
            }
        }
    };
    dfs(0);

    cout<<ans<<endl;
    REP(i,5) {
        cout<<task2[i]<<endl;
    }
    return 0;
}
0