結果

問題 No.1470 Mex Sum
コンテスト
ユーザー Kou0406
提出日時 2025-01-12 15:18:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 800 bytes
コンパイル時間 740 ms
コンパイル使用メモリ 65,988 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2025-01-12 15:18:32
合計ジャッジ時間 4,336 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 10 WA * 39
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:12:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   12 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
main.cpp:16:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   16 |         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,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("%d\n",ans);
    return 0;
}
0