【算法题】动态规划-旅行

2021-04-29 09:01:32 阅读：186 来源： 互联网

题目描述

在这里插入图片描述
示例1
输入：

4 2 7
1 2 3 4

输出：

说明：

2到7的合法路径有：
2->1->5->6->7，长度为4
2->3->4->8->7，长度为7
2->3->7，长度为4
2->6->7，长度为3
长度和为4+7+4+3=18

链接：https://ac.nowcoder.com/acm/contest/12949/F
来源：牛客网
思路解析：这是今天下午参加牛客比赛的最后一题，比较有意思。看起来是个图论相关的，仔细分析所以动态规划，里面的转态转移要搞懂。题目的意思是找出从点S到点t的所有不重复且没有重复点的道路，然后累加计算路径和。所以我们将所以的路径都找出来，如果简单的使用广搜加判重，代码结构与时间复杂度会很高，有许多不符合要求的转态会被搜索到，所以使用动态规划才是这道题的正解。
使用动态规划第一要分析转态的转移。它可能是下面的这种状态。
这时可以将它分成三部分：
第一部分：计算点s到点s+n的路径和与路径种类。这里只存在一些[型的路径。即
4-n4
4-3-n3-n4
4-3-2-n2-n3-n4
4-3-2-1-n1-n2-n3-n4
共4种，即种类数 f[s][1]=s;
路径和：g[s][1]=(a1+…+a4)+(1+1+2+2+3+3+4+4)=(a1+…+a4)+4*3;
第二部分：计算（s，t)存在二个方向的状态转移；
由图可发现，走到点n4有二个方向进行状态转移，同样走到点5有二个方向进行状态转移；即可由初始状态点4与点n4二个状态不同转移得到点t与点n(t-1)的状态。此时还还剩余点n(t+1)的状态未得到。
第三部分：[t，n]只存在一个方向的转态转移。
计算点n(t+1)，该状态用点出发t不断进行状态转移得到。此时要注意状态的组合产生新的状态，而不是单纯的转移。
最后：最终答案的nt状态可用点t、点n(t-1)与点n(t+1)得到
在这里插入图片描述
具体代码：

#include<bits/stdc++.h>
using namespace std;

#define mp make_pair
const int N=1e5+10;
typedef long long ll;
const ll M=998244353;  //余数
int n,s,t;
ll a[N];
ll g[N][2];//到达该点所有道路的和
ll f[N][2];//到达该点道路种类数

void solve()
{
    scanf("%d %d %d",&n,&s,&t);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    t-=n;  //转换为表第二行的序号
    //思路：将网格图分成三部分：
    //第一部分：[0,s]只存在一些[型的路；
    //第二部分：（s，t)存在二个方向的状态转移；
    //第三部分：[t，n]只存在一个方向的转态转移。
    ll sum=0;
    for(int i=1;i<=s;i++) sum+=a[i];
    f[s][1]=s;
    g[s][1]=(sum+(s-1ll)*s)%M;     //第一部分：[0,s]只存在一些[型的路；
    f[s][0]=1;
    g[s][0]=0;
    for(int i=s+1;i<t;i++) {    //第二部分：（s，t)存在二个方向的状态转移；
        f[i][0]=(f[i-1][0]+f[i-1][1])%M;
        g[i][0]=(g[i-1][0]+f[i-1][0]+g[i-1][1]+(1+a[i])*f[i-1][1]%M)%M;
        f[i][1]=(f[i-1][0]+f[i-1][1])%M;
        g[i][1]=(g[i-1][1]+f[i-1][1]+g[i-1][0]+(1+a[i])*f[i-1][0]%M)%M;
    }
    ll ans=f[t-1][1]+g[t-1][1];
    ans%=M;
    sum=0;
    //第三部分：[t，n]只存在一个方向的转态转移。
    for(int i=t;i<=n;i++) sum+=a[i];
    ans+=(f[t-1][0]*(n-t+1+sum%M+(n-t)*(n-t+1ll)%M)%M+g[t-1][0]*(n-t+1)%M)%M;   
    printf("%lld\n",ans%=M);
}

int main()
{
    solve();
    return 0;
}

标签：旅行,路径,状态,ll,转态,算法,动态,n4,转移
来源： https://blog.csdn.net/qq_36553828/article/details/116105185

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

【算法题】动态规划-旅行

题目描述