PAT（甲级）2019年秋季考试

7-1 Forever (20分)

题意： $f(x)$ 表示 x 各个数位之和。给定 k、m，请你找到一个 k 位的数 A 满足：

• $f(A)=m$
• $f(A+1)=n$
• $gcd(m,n)>2$ 且 $gcd(m,n)$ 为素数

按照 n 和 A 从小到大输出 n、A

思路： dfs + 剪枝

#include <bits/stdc++.h>
#define fi first
#define se second
#define ll long long
using namespace std;
const int maxn=1e5+5;

int t,n,m,k,a;
vector<pair<int,int> > ans;

int getval(int n)
{
    int res=0;
    while(n)
    {
        res+=n%10;
        n/=10;
    }
    return res;
}
bool isPrime(int n)
{
    if(n<=1) return 0;
    for(int i=2; i*i<=n; ++i)
        if(n%i==0) return 0;
    return 1;
}
void dfs(int pos,int sum,ll val,bool lim)
{
    if(pos==0)
    {
        int n=getval(val+1);
        int gcd=__gcd(n,sum);
        if(sum==m&&gcd>2&&isPrime(gcd))
            ans.push_back({n,val});
        return;
    }
    if(sum>m||sum+pos*9<m) return;
    for(int i=lim; i<=9; ++i)
        dfs(pos-1,sum+i,val*10+i,0);
}

int main()
{
    scanf("%d",&t);
    for(int Case=1; Case<=t; ++Case)
    {
        scanf("%d%d",&k,&m);
        ans.clear();
        dfs(k,0,0,1);
        printf("Case %d\n",Case);
        int m=ans.size();
        if(m==0) puts("No Solution");
        else
        {
            sort(ans.begin(),ans.end());
            for(int i=1; i<=m; ++i) printf("%d %d\n",ans[i-1].fi,ans[i-1].se);
        }
    }
    return 0;
}

7-2 Merging Linked Lists (25分)

题意：将小链表向大链表合并。

思路：简单模拟题

#include <bits/stdc++.h>
#define fi first
#define  se second
#define ll long long
using namespace std;
const int maxn=1e5+5;
int l1,l2,n;
struct Node
{
    int val;
    int nxt;
} p[maxn];
vector<pair<int,int> > vec1,vec2,res;
int main()
{
    cin>>l1>>l2>>n;
    for(int i=1; i<=n; ++i)
    {
        int add,val,nxt;
        cin>>add>>val>>nxt;
        p[add]= {val,nxt};
    }
    for(int i=l1; i!=-1; i=p[i].nxt)
        vec1.push_back({i,p[i].val});
    for(int i=l2; i!=-1; i=p[i].nxt)
        vec2.push_back({i,p[i].val});
    if(vec1.size()<vec2.size()) swap(vec1,vec2);

    int l=0,r=vec2.size()-1;

    for(int i=1; i<=n; ++i)
    {
        if(i%3==0&&r>=0) res.push_back(vec2[r--]);
        else if(l<=vec1.size()-1) res.push_back(vec1[l++]);
    }
    int m=res.size();
    if(m-1>=0)
    {
        for(int i=0; i<m-1; ++i)
            printf("%05d %d %05d\n",res[i].fi,res[i].se,res[i+1].fi);
        printf("%05d %d -1\n",res[m-1].fi,res[m-1].se);
    }
    return 0;
}

7-3 Postfix Expression (25分)

题意：给定一颗二叉树，输出后缀表达式。

思路：两个节点其中一个为空时，不输出输出中缀表达式即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=100;
int n;
string a[maxn];
int ls[maxn],rs[maxn],in[maxn];
struct Node
{
    string val;
    Node* ls,* rs;
};
Node* build(int u)
{
    if(u==-1) return NULL;
    Node* node=new Node;
    node->val=a[u];
    node->ls=build(ls[u]);
    node->rs=build(rs[u]);
    return node;
}
string ans;
void dfs(Node* root)
{
    if(root==NULL) return;
    ans+='(';
    if(root->ls!=NULL&&root->rs!=NULL)
    {
        dfs(root->ls);
        dfs(root->rs);
        ans+=root->val;
    }
    else
    {
        dfs(root->ls);
        ans+=root->val;
        dfs(root->rs);
    }
    ans+=')';
}
int main()
{
    cin>>n;
    for(int i=1; i<=n; ++i)
    {
        cin>>a[i]>>ls[i]>>rs[i];
        in[ls[i]]++;
        in[rs[i]]++;
    }
    int s=-1;
    for(int i=1; i<=n; ++i)
        if(in[i]==0) s=i;
    Node* root=build(s);
    dfs(root);
    cout<<ans<<"\n";
    return 0;
}

7-4 Dijkstra Sequence (30分)

题意：给定一个图，给定一些序列，判断是否是 dijstra 转移过程中可能会选择的序列。

思路：直接在 dijstra 转移的过程中，判断能不能用给定的方案转移

#include <bits/stdc++.h>
#define fi first
#define se second
#define ll long long
using namespace std;
const int maxn=1e5+5,inf=1e9;

int n,m;
int dis[maxn],visit[maxn],seq[maxn];
vector<pair<int,int> > e[maxn];

bool dijstra()
{
    for(int i=1; i<=n; ++i) dis[i]=inf,visit[i]=0;
    int s=seq[1];
    int now=1;
    dis[s]=0;
    for(int i=1; i<=n-1; ++i)
    {
        int mi=inf;
        for(int j=1; j<=n; ++j)
            if(!visit[j]&&mi>dis[j]) mi=dis[j];
        //printf("i=%d mi=%d\n",i,mi);
        if(mi!=dis[seq[now]]) return 0;
        visit[seq[now]]=1;
        int u=seq[now];
        now++;
        for(auto x: e[u])
        {
            int v=x.fi,w=x.se;
            if(dis[v]>dis[u]+w)
                dis[v]=dis[u]+w;
        }
    }
    return 1;
}

int main()
{
    cin>>n>>m;
    for(int i=1; i<=m; ++i)
    {
        int u,v,w;
        cin>>u>>v>>w;
        e[u].push_back({v,w});
        e[v].push_back({u,w});
    }
    int k;
    cin>>k;
    while(k--)
    {
        for(int i=1; i<=n; ++i) scanf("%d",&seq[i]);
        puts(dijstra()?"Yes":"No");
    }
    return 0;
}