NOIP 2016 换教室

---

【分析】
一个括号引发的惨案…
码风突变…凑乎着看…心累累

为大家分享wxl的一句话：
你只能决定决策，不能决定结果orz
dp[i][j][k]表示前i个时间段，换了j次教室，k表示第i时间段换不换…k不能表示为当前是选的c还是选的d
受益匪浅（手动再见）

---

【代码】

//NOIP 2016 换教室 
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define inf 1e9+7
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
using namespace std;
const int mxn=2005;
int n,m,p,w;   //可以换p节课,有w个时间段 
int dis[305][305],c[mxn],d[mxn];
int vis[mxn][mxn][2];
double dp[mxn][mxn][2],K[mxn];
inline void init()
{
    scanf("%d%d%d%d",&w,&p,&n,&m);
    memset(dis,0x3f,sizeof dis);
    fo(i,1,w) scanf("%d",&c[i]);
    fo(i,1,w) scanf("%d",&d[i]);
    fo(i,1,w) scanf("%lf",&K[i]);
    while(m--)
    {
        int u,v,x;
        scanf("%d%d%d",&u,&v,&x);
        dis[u][v]=dis[v][u]=min(dis[u][v],x);
    }
    fo(k,1,n)
      fo(i,1,n)
        fo(j,1,n)
          dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
    fo(i,0,n) dis[0][i]=dis[i][0]=dis[i][i]=0;
}
inline double dfs(int now,int num,int flag)  //在now位置，换了num节课的期望路程,flag是当前位置是否换 
{
    if(now==w) return 0;
    if(vis[now][num][flag]) return dp[now][num][flag];
    if(!flag)
    {
        if(num<p)
          dp[now][num][0]=dfs(now+1,num+1,1)+dis[c[now]][d[now+1]]*K[now+1]   //success
                         +dis[c[now]][c[now+1]]*(1-K[now+1]);  //fail 
        else dp[now][num][0]=inf;
        dp[now][num][0]=min(dp[now][num][0],dfs(now+1,num,0)+dis[c[now]][c[now+1]]);
    }
    else
    {
        if(num<p)
        dp[now][num][1]=dfs(now+1,num+1,1)
                   +dis[c[now]][d[now+1]]*(1-K[now])*K[now+1] //success
                   +dis[d[now]][c[now+1]]*K[now]*(1-K[now+1])    //fail 
                   +dis[c[now]][c[now+1]]*(1-K[now])*(1-K[now+1])
                   +dis[d[now]][d[now+1]]*K[now]*K[now+1];  
        else dp[now][num][1]=inf;
        dp[now][num][1]=min(dp[now][num][1],dfs(now+1,num,0)+dis[c[now]][c[now+1]]*(1-K[now])
                                           +dis[d[now]][c[now+1]]*K[now]);
    }
    vis[now][num][flag]=1;
    return dp[now][num][flag];
}
int main()
{
//  freopen("classrooma.in","r",stdin);
//  freopen("classrooma.out","w",stdout);
    init();
    memset(vis,0,sizeof vis);
    double ans=dfs(0,0,0);
    printf("%.2lf\n",ans+1e-4);
    return 0;
}