結果

問題 No.1440 The Quiz Competition
ユーザー KKT89
提出日時 2021-03-26 23:07:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,222 bytes
コンパイル時間 2,208 ms
コンパイル使用メモリ 195,140 KB
最終ジャッジ日時 2025-01-19 23:27:18
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 3 WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
typedef unsigned long long ull;

mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
ll myRand(ll B) {
	return (ull)rng() % B;
}

int main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	int q; cin >> q;
	while(q--){
		ll n,a,w,k; cin >> n >> a >> w >> k;
		if(k==1){ // 多分OK
			if(a)printf("%lld\n",a);
			else{
				if(w<n-1){
					printf(":(\n");
				}
				else{
					printf("0\n");
				}
			}
		}
		else if(n==k){ 
			if(w==0){ // ここも多分OK
				if(a<n-1){
					printf(":(\n");
				}
				else{
					printf("0\n");
				}
			}
			else{
				ll L=-1,R=w; // 最下位が不正解する回数
				while(R-L>1){
					ll mid=(L+R)/2;
					ll tmp=(w-mid)/(n-1);
					ll m=tmp*(tmp+1)/2*(n-1);
					ll u=w-tmp-tmp*(n-1);
					ll uu=(n-1)-u;
					ll aa=((mid*(mid+1)/2-1)>=m+(tmp+1)?0:(mid*(mid+1)/2-1)-m-(tmp+1));
					aa*=u;
					ll aaa=((mid*(mid+1)/2-1)>=m?0:(mid*(mid+1)/2-1)-m);
					aaa*=uu;
					if(aa+aaa<=a){
						R=mid;
					}
					else{
						L=mid;
					}
				}
				ll pi=(w+n-1)/n;
				L=max(R,pi);
				ll ans=-1e18;
				auto calc=[&](ll val,ll rem)->ll{
					ll b=(w-val)/(n-1); ll bb=b+1;
					ll cc=w-val-b*(n-1); ll c=n-1-cc;
					b=b*(b+1)/2,bb=bb*(bb+1)/2;
					val=val*(val+1)/2;
					ll res=-val;
					ll u=min(max(0LL,val-bb-1),rem);
					rem-=u;
					val+=u; res+=u;
					u=min(max(0LL,val-b-1),rem/(cc+1));
					res+=u; val+=u;
					rem-=u*(cc+1);
					res+=rem/n;
					return res;
				};
				R=w;
				for(int i=0;i<50 and L+100<R;i++){
					ll nl=(2*L+R)/3,nr=(L+2*R)/3;
					if(calc(nl,a)<calc(nr,a)){
						L=nl;
					}
					else{
						R=nr;
					}
				}
				for(int i=L;i<=L+100 and i<=w;i++){
					ans=max(ans,calc(i,a));
				}
				printf("%lld\n",ans);
			}
		}
		else{
			if(a==0){ // ここもOK
				if(w>=1+(n-k)){
					printf("-1\n");
				}
				else{
					printf(":(\n");
				}
			}
			else if(k-1<=a){ // ここも多分OKなはず…
				ll L=0,R=2e9;
				while(R-L>1){
					ll mid=(L+R)/2;
					if((k-1)*(mid+1)+mid<=a)L=mid;
					else R=mid;
				}
				printf("%lld\n",L);
			}
			else{
				if(w<n-k+1){
					printf(":(\n");
				}
				else{
					printf("-1\n");
				}
			}
		}
	}
}
0