题目
参考了带lao的题解https://www.acwing.com/solution/content/110229/
题意: 给定n个数,要把数组中的所有元素变为1。你有魔法,可以把任意连续个相同的数x变成sqrt(x/2+1)。求至少用多少次魔法
思路: 显然,先对大的数使用魔法更优,有机会合并。
可以用暴力,每个数变5、6次就变成1了,不看/2,开根号也就6次,有/2更快。可以记录每个数变到1的路径,从小到大。把本题逆过来,数组中所有数都为1,变成输入的数组。以样例为例,只要最后一次是2变成1的并且连续的,在路径中都是第一步,这样只记录一次贡献足以。当所有数的路径都走完,就下班了。想到了变化次数不会很多,但是没想到这么妙的做法。
时间复杂度: O(n)
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 2e5+10;
#define mem(a,x) memset(a,x,sizeof(a))
#define fir(i,a,b) for(int i=a;i<=b;++i)
int n,m,k,T;
vector<ll> va[N];
ll a[N];
void solve()
{
cin>>n;
for(int i=1;i<=n;++i)
{
cin>>a[i];
stack<ll> st;
ll x = a[i];
while(x != 1)
{
st.push(x);
x = sqrt(x/2+1);
}
while(st.size())
{
va[i].push_back(st.top());
st.pop();
}
}
ll ans = 0;
int now = 0; //表示当前的层数
while(1)
{
int cnt = 0;
for(int i=1;i<=n;++i)
{
if(va[i].size() <= now) {cnt++;continue;}
if(now < va[i-1].size())
{
if(va[i][now] != va[i-1][now]) ans++;
}
else
{
ans++;
}
}
now++;
if(cnt==n) break;
}
cout<<ans;
}
int main(void)
{
ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
solve();
return 0;
}