cf D. Valera and Fools

http://codeforces.com/contest/369/problem/D

标号最小的两个人会有四种状态：a活b活，a死b活，a活b死，a死b死；按照这四种状态dfs就可以求出最后的数量。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define maxn 5000
 using namespace std;

 int n,k;
 int ans;
 int p[maxn];
 bool vis[maxn][maxn];
 int c[maxn];

 void dfs(int num1,int num2,int cnt)
 {
     if(vis[num1][num2]) return ;
     vis[num1][num2]=true;
     ans++;
     if(num2>n||cnt==k) return;
     if(p[num1])
     {
         if(c[num2]) dfs(num2+,num2+,cnt+);
         if(c[num2]<) dfs(num1,num2+,cnt+);
     }
     if(p[num1]!=)
     {
         if(c[num2]) dfs(num2,num2+,cnt+);
     }
 }

 int main()
 {
     while(scanf("%d%d",&n,&k)!=EOF)
     {
         ans=;
         memset(vis,false,sizeof(vis));
         memset(p,,sizeof(p));
         memset(c,,sizeof(c));
         for(int i=; i<=n; i++)
         {
             scanf("%d",&p[i]);
         }
         for(int i=n; i>=; i--)
         {
             c[i]=max(c[i+],p[i]);
         }
         dfs(,,);
         printf("%d\n",ans);
     }
     return ;
 }