Codeforces369 D. Valera and Fools（dp）

题意：

N个人，要进行K轮游戏。
每个人都有一把枪，每个人一开始都将抢指向在场上除自己以外编号最小的那个人身上。（即编号最小的那个人会将枪口指向编号次小的那个人身上）

已知每个人打中的概率P[i]，问K轮内，会有几种存活情况出现。

数据范围：
n,k<=3000
0<=P[i]=100

解法：

d[t][i][j]表示前t轮，存活的最小编号为i，次小编号为j的方案是否存在.
转移:
枚举i,j,k.
令di为i被杀死的概率, dj为j被杀死的概率,然后转移方程即可.
需要考虑概率=0和概率=100的情况.

但是状态数量太多了，复杂度O(n^3)，需要优化.
可以用d[i][j]表示c存活的最小编号为i，次小编号为j，出现的最小回合数。
转移方式类似，但是复杂度就降到O(n^2)了.

Code：

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define PI pair<int, int>
const int maxm=2e5+5;
const int mod=1e9+7;
int n,k;
int d[3333][3333];
int p[maxm];
int sub[maxm];
void solve(){
  cin>>n>>k;
  for(int i=1;i<=n;i++)cin>>p[i];
  if(n==1){
    cout<<1<<endl;return ;
  }
  for(int i=1;i<3333;i++){
    for(int j=1;j<3333;j++){
      d[i][j]=1e18;
    }
  }
  for(int i=n;i>=1;i--){
    sub[i]=max(sub[i+1],p[i]);
  }
  d[1][2]=0;
  for(int i=1;i<=n;i++){
    for(int j=i+1;j<=n;j++){
      if(d[i][j]>=k)continue;
      int di=sub[j];
      int dj=p[i];
      if(di==0&&dj==0)continue;
      // i和j一起死
      if(di>0&&dj>0)d[j+1][j+2]=min(d[j+1][j+2],d[i][j]+1);
      // i死j没死
      if(di>0&&dj<100)d[j][j+1]=min(d[j][j+1],d[i][j]+1);
      // i没死j死
      if(di<100&&dj>0)d[i][j+1]=min(d[i][j+1],d[i][j]+1);
    }
  }
  int ans=0;
  // 剩下至少两个
  for(int i=1;i<=n;i++){
    for(int j=i+1;j<=n;j++){
      if(d[i][j]<=k)ans++;
    }
  }
  // 剩下一个
  for(int i=1;i<=n;i++){
    if(d[i][n+1]<=k)ans++;
  }
  // 剩下零个
  if(d[n+1][n+2]<=k)ans++;
  cout<<ans<<endl;
}
signed main() {
// #define MULTI_CASE
  ios::sync_with_stdio(0);
  cin.tie(0);
#ifndef ONLINE_JUDGE
  freopen("../in.txt", "r", stdin);
  freopen("../out.txt", "w", stdout);
#endif
#ifdef MULTI_CASE
  int T;
  cin >> T;
  while (T--)
#endif
    solve();
  return 0;
}

---