結果

問題 No.1470 Mex Sum
ユーザー zumin
提出日時 2021-05-01 15:13:24
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 42 ms / 2,000 ms
コード長 872 bytes
コンパイル時間 530 ms
コンパイル使用メモリ 66,688 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-19 23:07:30
合計ジャッジ時間 4,751 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:37:21: warning: 'mex' may be used uninitialized [-Wmaybe-uninitialized]
   37 |             ans+=mex*cnt[i]*cnt[j];
      |                  ~~~^~~~~~~
main.cpp:27:28: note: 'mex' was declared here
   27 |             ll mexl[4]={0},mex;
      |                            ^~~

ソースコード

diff # 

#include <iostream>
#include <vector>

using namespace std;
using ll=long long;

int main(){
    ll n;
    cin>>n;
    ll cnt[3]={0};
    ll ans=0;
    
    for(ll i=0;i<n;i++){
        ll a;
        cin>>a;
        if(a==1){
            cnt[0]++;
        }else if(a==2){
            cnt[1]++;
        }else{
            cnt[2]++;
        }
    }

    for(int i=0;i<3;i++){
        for(int j=i+1;j<3;j++){
            ll mexl[4]={0},mex;
            mexl[0]=1;
            mexl[i+1]++;
            mexl[j+1]++;
            for(int k=1;k<4;k++){
                if(mexl[k]==0){
                    mex=k;
                    break;
                }
            }
            ans+=mex*cnt[i]*cnt[j];
        }
        if(i==0){
            ans+=2*cnt[i]*(cnt[i]-1)/2;
        }else{
            ans+=cnt[i]*(cnt[i]-1)/2;
        }
    }
    cout<<ans<<endl;
    return 0;
}
0