hdu5943素数间隙与二分匹配

#include <cstdio>
#include <iostream>
#include<cstring>
using namespace std;
#define ll long long
const int maxn=1000+10;
int f[maxn],to[maxn*maxn],nex[maxn*maxn],cnt,vis[maxn],pe[maxn],n,s;
void add(int a,int b)
{
    cnt++;
    to[cnt]=b;
    nex[cnt]=f[a];
    f[a]=cnt;
}
bool is_ok(int x)
{
    for(int i=f[x]; i; i=nex[i])
    {
        if(vis[to[i]]==0)
        {
            vis[to[i]]=1;
            if(pe[to[i]]==0||is_ok(pe[to[i]]))
            {
                pe[to[i]]=x;
                return true;
            }
        }
    }
    return false;
}
int main()
{
    int T;
    cin>>T;
    for(int cn=1; cn<=T; cn++)
    {
        cnt=0;
        scanf("%d %d",&n,&s);
        if(n>s)
        {
            int kk=n;
            n=s;
            s=kk;
        }
        if(n>1000)
        {
            printf("Case #%d: No\n",cn);
            continue;
        }
        for(int i=1; i<maxn; i++)
            f[i]=0,pe[i]=0;
        memset(nex,0,sizeof(nex));
        for(ll i=s+1; i<=n+s; i++)
        {
            for(int j=1; j<=n; j++)
            {
                if(i%j==0)
                    add(j,i-s);
            }
        }
        int ans=0;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)vis[j]=0;
            if(is_ok(i))ans++;
        }
        if(ans!=n)
            printf("Case #%d: No\n",cn);
        else
            printf("Case #%d: Yes\n",cn);
    }
    return 0;
}