Project Euler problem 61

题意很明了。

然后我大概的做法就是暴搜了

先把每个几边形数中四位数的处理出来。

然后我就DFS回溯着找就行了。

比较简单吧。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
using namespace std;
vector<int>g[7], v[7][10000];
int vis[7];
void get_three()
{
    for(int i = 1; ; i++)
    {
        int x = i * (i + 1) / 2;
        if(x >= 10000) break;
        if(x >= 1000) g[0].push_back(x);
    }
}
void get_four()
{
    for(int i = 1; ; i++)
    {
        int x = i * i;
        if(x >= 10000) break;
        if(x >= 1000) g[1].push_back(x);
    }
}
void get_five()
{
    for(int i = 1; ; i++)
    {
        int x = i * (3 * i - 1) / 2;
        if(x >= 10000) break;
        if(x >= 1000) g[2].push_back(x);
    }
}
void get_six()
{
    for(int i = 1; ; i++)
    {
        int x = i * (2 * i - 1);
        if(x >= 10000) break;
        if(x >= 1000) g[3].push_back(x);
    }
}
void get_seven()
{
    for(int i = 1; ; i++)
    {
        int x = i * (i * 5 - 3) / 2;
        if(x >= 10000) break;
        if(x >= 1000) g[4].push_back(x);
    }
}
void get_eight()
{
    for(int i = 1; ; i++)
    {
        int x = i * (3 * i - 2);
        if(x >= 10000) break;
        if(x >= 1000) g[5].push_back(x);
    }
}
int a[7];
int ans;
int flag;
void dfs(int deep)
{
    if(flag) return;
    if(deep == 6)
    {
        if(a[5] % 100 == a[0] / 100)
        {
            flag = 1;
            printf("%d
", a[0] + a[1] + a[2] + a[3] + a[4] + a[5]);
            return;
        }
    }
    for(int i = 0; i < 6; i++)
    {
        if(!vis[i])
        {
            for(int j = 0; j < g[i].size(); j++)
            {
                int y = g[i][j];
                if(!deep || y / 100 == a[deep - 1] % 100)
                {
                    vis[i] = 1;
                    a[deep] = y;
                    dfs(deep + 1);
                    vis[i] = 0;
                }
            }
        }
    }
}
int main()
{
    get_three();
    get_four();
    get_five();
    get_six();
    get_seven();
    get_eight();
    dfs(0);
    return 0;
}