codeforce447 D SGU 548 贪心+优先队列

codeforce447 D - DZY Loves Modification 
题意：
有一个n*m的矩阵，每次可以选择一行或者一列，可以得到这行或这列的所有元素sum的积分，
然后使这一列/行的每一个元素都减少p，接着再选择一行或一列，共操作k次，n,m<=1000,k<=1000000
求最多能得到多少积分

思路：
每次选择sum最大的行或者列，但是时间复杂度太大，过不去~
若选择一行，则每个列的sum一定减少p，同理；
若选择的行数和列数确定下来了，选择i行，k-i列，那么行和列之间谁选选择就没有影响；
因为在选择行的时候，每一列都的sum都减少p，每一列的相对大小没有变化，前k-i大的列还是那几列，同列。
所以可以用优先队列O（n）预处理出row_sum[i]和col_sum[k-i]的值。选i行则消耗了每列的i*p的大小，共k-i列，故共消耗了i*(k-i)*p，答案为
max( row_sum[i], col_sum[k-i]）- i*(k-i)*p

注意：！！！！！WA了很久T.T，因为i*(k-i)*p会超过int范围！！！

/*===============================================================
*   Copyright (C) 2014 All rights reserved.
*   
*   File Name: codeforces447_D_greedy.cpp
*   Author:sunshine
*   Created Time: 2014年07月23日
*
================================================================*/
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>

using namespace std;

int arr[1001][1001];
long long row_sum[1000001];
long long col_sum[1000001];

int main(){
    int n,m,k,p;
    cin >> n >> m >> k >> p;
        
    priority_queue<int>row,col;
        
    for(int i = 0;i < n;i ++){
        long long tmp =  0;
        for(int j = 0;j < m;j ++){
            cin >> arr[i][j];
            tmp += arr[i][j];
        }
        row.push(tmp);
    }

    for(int j = 0;j < m;j ++){
        long long tmp = 0;
        for(int i = 0;i < n;i ++){
            tmp += arr[i][j];
        }
        col.push(tmp);
    }
        
    row_sum[0] = 0;
    for(int i = 1;i <= k;i ++){
        row_sum[i] = row_sum[i - 1] + row.top();
        row.push(row.top() - m * p);
        row.pop();
    }

    col_sum[0] = 0;
    for(int i = 1;i <= k;i ++){
        col_sum[i] = col_sum[i-1] + col.top();
        col.push(col.top() - n * p);
        col.pop();
    }

    long long res = row_sum[0] + col_sum[k];
    for(int i = 1;i <= k;i ++){
        res = max(res, row_sum[i] + col_sum[k - i] - (long long)i * (k - i) * p);
    }
    cout << res << endl;
    return 0;
}

SGU 548 Dragons and Princesses
题意：
唐僧从起点1出发，依次从1到2到3……到n，第n位一定是一位公主，在2~n-1中可能是公主，也可能是龙，
d表示龙，p表示公主，每只龙有di个金币，每位公主有pi的美丽值，杀死一只龙就可以得到相应的财富值。
唐僧喜欢位置为n的那位公主，所以要和位置为n的公主结婚，所以他不能被其他公主喜欢上，必须被最后一个公主喜欢。
他被公主喜欢上的条件是：在遇到位置为i的公主是，如果pi小于等于唐僧杀死的龙的数量，那么公主
就认为唐僧很V5，会喜欢上唐僧，问题，唐僧成功和位置为n的公主结婚的情况下，最多能得到多少金币。
如果不能结婚，输出-1。

思路：
利用优先队列遇到龙就吃掉，遇到公主，发现吃多了，就把多余的小的都吐出来，当然遇到最后一个公主的时候不需要吐出来。

#include <stdio.h>
#include <iostream>
#include <queue>
#include <algorithm>

using namespace std;

struct node{
    int key;
    int id;
    friend bool operator < (node a,node b){
        return a.key > b.key;
    }
}dro;

int arr[200005];

int main(){
    int n;
    char ch[10];
    int key;
    int base;
    while(scanf("%d", &n) != EOF){
        
        priority_queue<node>que;

        for(int i = 0;i < n - 1;i ++){
            scanf("%s %d", ch, &key);

            if(ch[0] == 'd'){
                dro.key = key;
                dro.id = i + 2;
                que.push(dro);
            }
            if(ch[0] == 'p'){
                if(i != n - 2){
                    while(que.size() >= key){
                        que.pop();
                    }
                }else{
                    base = key;
                }
            }
        }


        if(que.size() < base){
            puts("-1");
        }else{
            int res = 0;
            int top = 0;
            while(que.size()){
                dro = que.top();
                que.pop();
            //    printf("key:%d id:%d
",dro.key,dro.id);
                res += dro.key;
                arr[top++] = dro.id;
            }

            sort(arr,arr + top);

            printf("%d
%d
",res,top);

            for(int i = 0;i < top - 1;i ++){
                printf("%d ",arr[i]);
            }
            printf("%d
",arr[top - 1]);
        }
    }
    return 0;
}