问题 F: 小Z切木板

题目描述

小Z是个特别堕落的小朋友，上数学课的时候经常有奇奇怪怪的想法。
小Z的的包里有块nm的长方形木板，小Z想把木板全部切成11的小正方形（用来玩拼图）。但木板本身并不均匀，因此从不同的线切割下去要花不同的代价。而且对于一块木板，切割一次以后就被分割成两块，而且由于不能把这两块木板拼在一起然后一刀切成四块（动静太大），所以只能两块分别再进行一次切割。现在小Z并不想花太多的力气，所以他来找你帮他求一求最小的代价。

输入

第一行包括N和M（1≤N，M≤1000），表示长N宽M的矩阵。
第二行包括N-1个非负整数，分别表示沿着N-1条横线切割的代价。
第三行包括M-1个非负整数，分别表示沿着M-1条竖线切割的代价。

输出

输出一行一个整数，表示最小切割代价。

样例输入 Copy

2 2
3
3

样例输出 Copy

9

提示

对于60%的数据：n≤100；
对于100%的数据：n≤1000，切割代价<=1000。

思路

按切割代价由大到小排序，因为目标是1*1的小木块，所以一定要切的嘛，然后横着切一刀以后，竖方向的就多了1，竖着切一刀同理，那就代价大的那刀先切。这样就会使总代价比较小了。

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
struct vode
{
    int ai,fla;
}ci[10005];
bool cmp(vode q,vode w)
{
    return q.ai>w.ai;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m,len=0;
    cin>>n>>m;
    for(int i=1;i<=n-1;i++){
        cin>>ci[++len].ai;
        ci[len].fla=0;
    }
    for(int i=1;i<=m-1;i++){
        cin>>ci[++len].ai;
        ci[len].fla=1;
    }
    sort(ci+1,ci+1+len,cmp);
    ll ans=0;
    int num_x=1,num_y=1;
    for(int i=1;i<=len;i++){
        if(ci[i].fla==0){
            ans+=ci[i].ai*num_y;
            num_x++;
        }
        else{
            ans+=ci[i].ai*num_x;
            num_y++;
        }
    }
    cout<<ans;
    return 0;
}