2021 GDCPC广东省大学生程序设计竞赛 J Jerry（bfs/dp）

2021 GDCPC广东省大学生程序设计竞赛 J Jerry

题意：有q次询问，每次给定一个数，对于数x可以进行的操作是减去一个完全平方数或者加上一个完全平方数（进行操作后的数不能小于0或大于1e5）
解法1：bfs，先搜到的就是最少的操作数

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int cnt;
int st[N],dist[N];
int a[N];
int q[N];
void inint()
{
    for (int i=1;i*i<=1e5;i++) a[cnt++]=i*i;
}
void bfs()
{
    int hh=0,tt=1;
    st[0]=1;
    q[hh]=0;
    while(hh<tt)
    {
        int t=q[hh++];
        for (int i=0;i<cnt;i++)
        {
            if (t+a[i]<=1e5 && !st[t+a[i]])
            {
                st[t+a[i]]=1;
                dist[t+a[i]]=dist[t]+1;
                q[tt++]=t+a[i];
            }
            if (t-a[i]>=0 && !st[t-a[i]])
            {
                st[t-a[i]]=1;
                dist[t-a[i]]=dist[t]+1;
                q[tt++]=t-a[i];
            }
        }
    }
}
int main()
{
    int T;
    inint();
    bfs();
    cin>>T;
    while(T--)
    {
        int x;
        cin>>x;
        cout<<dist[x]<<endl;
    }
    return 0;
}

解法2：完全背包，假设每个完全平方数（正跟负）有无限个，先按照正序计算+1个完全平方数，再倒序计算-1个完全平方数

#include<bits/stdc++.h>
using namespace std;
const int N=1e6;
const int INF=0x3f3f;
int cnt;
int a[N];
int f[N];
void inint()
{
    memset(f,INF,sizeof(f));
    for (int i=1;i*i<=1e5;i++) 
    {
        a[cnt++]=i*i;
        f[i*i]=1;
    }
}
void dp()
{
    f[0]=0;
    for (int i=0;i<cnt;i++)
    {
        int v=a[i];
        for (int j=v;j<=1e5;j++)
            f[j]=min(f[j-v]+1,f[j]);
        for (int j=1e5;j>=0;j--)
            f[j]=min(f[j+v]+1,f[j]);
    }
}
int main()
{
    int T;
    inint();
    dp();
    cin>>T;
    while(T--)
    {
        int x;
        cin>>x;
        cout<<f[x]<<endl;
    }
    return 0;
}