1900. Brainwashing Device

http://acm.timus.ru/problem.aspx?space=1&num=1900

题目大意：

有N个车站，相邻车站之间形成一个段，这样就有N-1个段，每个段最多可以放一个洗脑的仪器，这样的话，所有经过这个段（放了仪器）的

人都会开心，我们有K个仪器，问怎么放可以让最多的人快乐

思路：

dp[i][j] 代表第i个仪器放在第j个段上的最优值，更新最优值时，需要枚举上一个仪器放的位置，在知道上一个仪器放的位置的情况下，可以

知道当前位置仪器可以作用到多少人。

注意人数全为0的情况

代码：

#include<iostream>
#include<stack>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<string>
#include<cmath>

using namespace std;

typedef long long ll;
typedef pair<int,int> pp;
const int INF=0x3f3f3f3f;
const int N=503;
int d[N][N];
short int f[N][N];
short int a[N][N];
int b[N][N];
int main()
{
    //freopen("data.in","r",stdin);
    int n,k;
    cin>>n>>k;
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    for(int i=1;i<n;++i)
    for(int j=i+1;j<=n;++j)
    cin>>a[i][j];
    for(int j=1;j<n;++j)
    for(int i=j;i<n;++i)
    {
        b[i][j]=b[i-1][j];
        for(int l=j;l<i;++l)
        b[i][j]-=a[l][i];
        for(int l=i+1;l<=n;++l)
        b[i][j]+=a[i][l];
    }
    memset(d,-1,sizeof(d));
    d[0][0]=0;
    memset(f,-1,sizeof(f));
    for(int i=0;i<k;++i)
    for(int j=i;j<n;++j)
    if(d[i][j]!=-1)
    for(int l=j+1;l<n;++l)
    if(d[i][j]+b[l][j+1]>d[i+1][l])
    {
        d[i+1][l]=d[i][j]+b[l][j+1];
        f[i+1][l]=j;
    }
    int x=k,y,ans=-1;
    for(int j=k;j<n;++j)
    if(d[k][j]>ans)
    {ans=d[k][j];y=j;}
    vector<int>vt;
    while(x!=0)
    {
        vt.push_back(y);
        y=f[x][y];
        --x;
    }
    sort(vt.begin(),vt.end());
    cout<<ans<<endl;
    for(unsigned int i=0;i<vt.size();++i)
    {
        if(i>0) cout<<" ";
        cout<<vt[i];
    }
    cout<<endl;
    return 0;
}