[DFS][记忆化][ZZNUOJ]生活所迫

一堆分两堆

平分代价一

否则代价二

求总代价

#pragma GCC optimize(2)
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <string>
#include <cstring>
#include <algorithm>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <ctime>
#include <vector>
#include <fstream>
#include <list>
#include <iomanip>
#include <numeric>
using namespace std;
typedef long long ll;
 
const int MAXN = 1e6 + 10;
 
ll ans = 0;
 
int arr[MAXN];


ll dfs(int n)
{
    if(n == 1)
        return 0;
    else if(arr[n] != 0)
        return arr[n];
    else
    {
        ll ans = 0;

        if(n % 2 == 0)
        {
            ans ++;
            
            ans += dfs(n / 2);
            ans += dfs(n / 2);
        }
        else
        {
            ans += 2;
 
            ans += dfs(n/2);
            ans += dfs(n - n/2);
        }
        return arr[n] = ans; //记忆化
    }
}
 
int main()
{
    ios::sync_with_stdio(false);
 
    cin.tie(0);
 
    cout.tie(0);
 
    memset(arr, 0, sizeof(arr));
    
    int T;
 
    cin>>T;
    
    while(T--)
    {
        arr[2] = 1;

        arr[1] = 0;
 
        ans = 0;
         
        int N;
 
        cin>>N;
 
        cout<<dfs(N)<<endl;
 
        //cout<<ans<<endl;
 
    }
     
    return 0;
}