結果

問題 No.1470 Mex Sum
コンテスト
ユーザー Kou0406
提出日時 2025-01-12 15:22:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 26 ms / 2,000 ms
コード長 810 bytes
コンパイル時間 636 ms
コンパイル使用メモリ 65,408 KB
実行使用メモリ 6,528 KB
最終ジャッジ日時 2025-01-12 15:22:08
合計ジャッジ時間 3,820 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:13:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   13 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
main.cpp:17:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   17 |         scanf("%d",&a[i]);
      |         ~~~~~^~~~~~~~~~~~

ソースコード

diff # 

#include<iostream>
#include<vector>
#define rep(i,n) for(i=0;i<(int)(n);i++)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;

int n;

int main(){
    int i;
    ll ans=0;
    scanf("%d",&n);
    vector<int> a(n),sum1(n+1),sum2(n+1),sum3(n+1);
    sum1[0]=sum2[0]=sum3[0]=0;
    rep(i,n){
        scanf("%d",&a[i]);
        sum1[i+1]=sum1[i]+(a[i]==1);
        sum2[i+1]=sum2[i]+(a[i]==2);
        sum3[i+1]=sum3[i]+(a[i]>=3);
    }
    rep(i,n){
        if(a[i]==1)
            ans+=2*(sum1[n]-sum1[i+1]+sum3[n]-sum3[i+1])+3*(sum2[n]-sum2[i+1]);
        else if(a[i]==2)
            ans+=sum3[n]-sum3[i+1]+sum2[n]-sum2[i+1]+3*(sum1[n]-sum1[i+1]);
        else
            ans+=sum3[n]-sum3[i+1]+sum2[n]-sum2[i+1]+2*(sum1[n]-sum1[i+1]);
    }
    printf("%lld\n",ans);
    return 0;
}
0