Xenia and Weights(dfs/dp)

https://codeforces.com/problemset/problem/339/C


思路:

看着确实可以dfs一波,但是这个dfs的复杂度是多少呢...翻了一下网上的其他题解都找到说这个事的。

如果有老哥懂大致的复杂度可以评论区指点一下我

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e3+100;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
int a[12];
char s[12];
bool flag=1;
LL ans[maxn];
void dfs(LL lw,LL rw,LL num,LL last,LL m){
    if(num>m||flag==0){
        flag=0;
        return;
    }
    if(num&1){
        bool f=0;
        for(LL i=1;i<=10;i++){
            if(i!=last&&a[i]){
                if(lw+i>rw){
                    f=1;
                    if(flag==1) ans[num]=i;
                    dfs(lw+i,rw,num+1,i,m);
                }
            }
        }
        if(f==0) return;
    }
    if(num%2==0){
        bool f=0;
        for(LL i=1;i<=10;i++){
            if(i!=last&&a[i]){
                if(rw+i>lw){
                    f=1;
                    if(flag==1) ans[num]=i;
                    dfs(lw,rw+i,num+1,i,m);
                }
            }
        }
        if(f==0) return;
    }
}
int main(void)
{
  scanf("%s",s+1);
  for(LL i=1;i<=10;i++){
     if(s[i]=='1') a[i]=i;
     else a[i]=0;
  }
  LL m;scanf("%lld",&m);
  dfs(0,0,1,-1,m);
  if(flag==1){
    printf("NO\n");
  }
  else{
    printf("YES\n");
    for(LL i=1;i<=m;i++){
        printf("%lld ",ans[i]);
    }
    printf("\n");
  }
return 0;
}

 

上一篇:孤荷凌寒自学python第八天 初识Python的序列之元组


下一篇:Python第八天——Json