C. Cow Program
题目链接
题意:起始时位于(1,0)点,每轮游戏有两个步骤:1.x+=a[x],y+=a[x]。2.x-=a[x],y+=a[x]。当x<=0或者x>n时停止。问最后y的值,如果不能停止,则输出-1.
思路:记忆化搜索。用一个val[x][2],来记录从x出发向左或者向右走的结果,如果再次搜到这个点直接返回值。并且用一个vis[x][2]来记录这个状态是否走过,如果走过,就说明有环,输出-1.
#include<bits/stdc++.h>
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=2e5+5;
LL a[maxn];
LL vis[maxn][2],val[maxn][2];
LL n;
LL dfs(LL x,LL op)
{
if(x<=0||x>n)return 0;
LL res=0;
if(val[x][op])return val[x][op];///这里一定要先判断val是否有值,再判断vis
if(vis[x][op])return val[x][op]=-1;
vis[x][op]=1;
LL num=1;
if(op==1)num=-1;
res=dfs(x+num*a[x],op^1);
if(res==-1)return -1;
return val[x][op]=res+a[x];
}
int main()
{
memset(val,0,sizeof(val));
memset(vis,0,sizeof(vis));
vis[1][0]=-1;
scanf("%d",&n);
for(LL i=2;i<=n;i++)
{
scanf("%I64d",&a[i]);
}
for(LL i=1;i<n;i++)
{
a[1]=i;
LL ans=dfs(i+1,1);
if(ans==-1)printf("-1\n");
else printf("%I64d\n",i+ans);
}
}
F - Password
题目链接
题意:求一个最长的子串,它即在开头出现,又在结尾出现,还在中间出现。
思路:KMP的next数组的应用。next数组表示的是当前这位之前与开头的最长公共前缀。如果next[i]=k,表示s[0] ~ s[k-1] =s[i-k] ~ s[i-1]。所以我们只需要先把中间的所有next的值都记录下来,然后从next[len]开始找,这个next值是否在中间出现过。
#include<bits/stdc++.h>
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=1e6+5;
char s[maxn];
int nex[maxn];
void getnext(char *ptr)///KMP的next数组模板
{
int i,n,k;
n=strlen(ptr);
k=-1;
nex[0]=-1;
i=0;
while(i<n)
{
if(k==-1 || ptr[i]==ptr[k])
{
i++;
k++;
nex[i]=k;
}
else k=nex[k];
}
}
int main()
{
scanf("%s",s);
getnext(s);
int len=strlen(s);
int ans=nex[len];
bool flag=0;
int x=len;
vector<int>v;
for(int i=1;i<len;i++)
{
v.push_back(nex[i]);
}
while(nex[x])
{
if(count(v.begin(),v.end(),nex[x]))///判断这个nex值是否在中间出现过
{
for(int i=0;i<nex[x];i++)///如果出现过,直接输出
{
printf("%c",s[i]);
}
printf("\n");
flag=1;
break;
}
x=nex[x];
}
if(!flag)
{
printf("Just a legend\n");return 0;
}
}