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[][];
 long long row_sum[];
 long long col_sum[];

 int main(){
     int n,m,k,p;
     cin >> n >> m >> k >> p;

     priority_queue<int>row,col;

     for(int i = ;i < n;i ++){
         long long tmp =  ;
         for(int j = ;j < m;j ++){
             cin >> arr[i][j];
             tmp += arr[i][j];
         }
         row.push(tmp);
     }

     for(int j = ;j < m;j ++){
         long long tmp = ;
         for(int i = ;i < n;i ++){
             tmp += arr[i][j];
         }
         col.push(tmp);
     }

     row_sum[] = ;
     for(int i = ;i <= k;i ++){
         row_sum[i] = row_sum[i - ] + row.top();
         row.push(row.top() - m * p);
         row.pop();
     }

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

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

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[];

 int main(){
     int n;
     char ch[];
     int key;
     int base;
     while(scanf("%d", &n) != EOF){

         priority_queue<node>que;

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

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

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

             sort(arr,arr + top);

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

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