結果

問題 No.412 花火大会
ユーザー fura
提出日時 2020-07-12 01:23:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 16 ms / 2,000 ms
コード長 874 bytes
コンパイル時間 1,991 ms
コンパイル使用メモリ 200,552 KB
最終ジャッジ日時 2025-01-11 19:52:09
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:26:21: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   26 |         int n; scanf("%d%d%d%d",&a[0],&a[1],&a[2],&n);
      |                ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:28:23: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   28 |         rep(i,n) scanf("%d",&b[i]);
      |                  ~~~~~^~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(n);i++)

using namespace std;

vector<int> a(3);

vector<int> solve(vector<int> b){
	int n=b.size();
	vector<int> cnt(8);
	rep(S,1<<n){
		int ok=0;
		rep(i,n) if(S>>i&1) {
			rep(j,3) if((ok>>j&1)==0 && a[j]<=b[i]) {
				ok|=1<<j;
				break;
			}
		}
		cnt[ok]++;
	}
	return cnt;
}

int main(){
	int n; scanf("%d%d%d%d",&a[0],&a[1],&a[2],&n);
	vector<int> b(n);
	rep(i,n) scanf("%d",&b[i]);

	sort(a.rbegin(),a.rend());
	sort(b.rbegin(),b.rend());

	auto cnt1=solve(vector(b.begin(),b.begin()+n/2));
	auto cnt2=solve(vector(b.begin()+n/2,b.end()));

	int ans=0;
	rep(S,8) rep(T,8) {
		bool ok=true;
		int carry=0;
		rep(i,3){
			int b1=S>>i&1;
			int b2=T>>i&1;
			if(b1+b2+carry==0){
				ok=false;
			}
			else{
				carry=b1+b2+carry-1;
			}
		}
		if(ok) ans+=cnt1[S]*cnt2[T];
	}
	printf("%d\n",ans);

	return 0;
}
0