「雜題亂刷」CF1759F

wangmarui發表於2024-05-23

題目連結

CF1759F All Possible Digits(luogu)

CF1759F All Possible Digits(codeforces)

題意簡述

有一個長度為 \(n\)\(p\) 進位制數,你需要求出至少透過幾次操作才可以讓 \(0 \sim p - 1\)\(p\) 個數字都至少出現過一遍(包括中間過程)。

解題思路

我們很容易就能發現答案是具有單調性的,也就是說,如果操作 \(x\) 次是合法的,那麼操作 \(x + 1\) 次也一定是合法的。

然後我們還可以發現一個性質,就是答案一定小於 \(p\),因為每一次操作都會使最後一位數字變化,所以操作 \(p - 1\) 次最後一位數字會變化 \(p\) 次,因此這個性質是正確的。

於是我們可以直接二分最少的操作次數即可,這裡我選擇使用分討數字操作後是否進位的方式來進行二分。

參考程式碼

#include<bits/stdc++.h>
using namespace std
//#define map unordered_map
#define forl(i,a,b) for(register long long i=a;i<=b;i++)
#define forr(i,a,b) for(register long long i=a;i>=b;i--)
#define forll(i,a,b,c) for(register long long i=a;i<=b;i+=c)
#define forrr(i,a,b,c) for(register long long i=a;i>=b;i-=c)
#define lc(x) x<<1
#define rc(x) x<<1|1
//#define mid ((l+r)>>1)
#define cin(x) scanf("%lld",&x)
#define cout(x) printf("%lld",x)
#define lowbit(x) (x&-x)
#define pb push_back
#define pf push_front
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define endl '\n'
#define QwQ return 0;
#define ll long long
#define ull unsigned long long
#define lcm(x,y) x/__gcd(x,y)*y
#define Sum(x,y) 1ll*(x+y)*(y-x+1)/2
#define aty cout<<"Yes\n";
#define atn cout<<"No\n";
#define cfy cout<<"YES\n";
#define cfn cout<<"NO\n";
#define xxy cout<<"yes\n";
#define xxn cout<<"no\n";
#define printcf(x) x?cout<<"YES\n":cout<<"NO\n";
#define printat(x) x?cout<<"Yes\n":cout<<"No\n";
#define printxx(x) x?cout<<"yes\n":cout<<"no\n";
ll t;
ll n,m;
ll a[110],b[110];
bool check(ll Mid)
{
	forl(i,0,n)
		a[i]=b[i];
	if(Mid+a[n]>=m)
	{
		map<ll,ll>mp;
		forl(i,1,n)
			mp[a[i]]++;
		ll pd=0;
		Mid-=m-a[n];
		ll lst=a[n]-1;
		a[n]=0,a[n-1]++;
		if(n-1==0)
			pd=1;
		forr(i,n-1,1)
		{
			if(a[i]==m)
			{
				a[i]=0,a[i-1]++;
				if(i==1)
					pd=1;
			}
		}
		if(pd)
			forl(i,0,n)
				mp[a[i]]++;
		else
			forl(i,1,n)
				mp[a[i]]++;
		while(mp[lst])
			lst--;
		return lst<=Mid;
	}
	else
	{
		map<ll,ll>mp;
		forl(i,1,n)
			mp[a[i]]++;
		ll lst=m-1;
		while(mp[lst])
			lst--;
		ll nxt=0;
		while(mp[nxt])
			nxt++;
//		cout<<Mid<<":"<<lst-a[n]<<endl;
		if(nxt<a[n])
			return 0;
		return max(0ll,lst-a[n])<=Mid;
	}
}
void solve()
{
	cin>>n>>m;
	forl(i,1,n)
		cin>>a[i],b[i]=a[i];
	ll L=0,R=m;
	while(L<R)
	{
		ll Mid=(L+R)/2;
		if(check(Mid))
			R=Mid;
		else
			L=Mid+1;
	}
	cout<<L<<endl;
}
int main()
{
	IOS;
	t=1;
	cin>>t;
	while(t--)
		solve();
	QwQ;
}