当前位置:编程学习 > C/C++ >>

hdu 4217 线段树

题目大意: t 组测试,每组测试有 n 个数(1~n),有 k 次循环,每次取出第k小的数,求所有取出的数字之和。


[cpp]
#include<iostream>  
#include<stdio.h>  
#include<string.h>  
#include<stdlib.h>  
#include<algorithm>  
#define SIZE 262145  
using namespace std; 
int array[SIZE<<1]; 
void creat(int left,int right,int root) 

    int m; 
    array[root] = right-left+1;//每个节点记录包含的数字个数  
    if(left == right){ 
        return; 
    } 
    m=(right+left)/2; 
    creat(left,m,root<<1);      //array[root]的左孩子为array[2*root]  
    creat(m+1,right,root<<1|1); //array[root]的右孩子为array[2*root+1]  

int Del(int left,int right,int root,int key) 

    int m,flag; 
    array[root]--; 
    if(left == right){ 
        return right; 
    } 
    m=(right+left)/2; 
    if(key<=array[root<<1]){//左子树的叶子节点数大于key  
        flag=Del(left,m,root<<1,key); 
    } 
    else{ 
        key-=array[root<<1];//在右子树中找,key要减去左子树包含的叶子节点数  
        flag=Del(m+1,right,root<<1|1,key); 
    } 
    return flag; 

int main() 

    int t,n,k,kk,i,j; 
    long long sum; 
    scanf("%d",&t); 
    for(i=1;i<=t;i++){ 
        sum=0; 
        scanf("%d%d",&n,&k); 
        creat(1,n,1); 
        for(j=0;j<k;j++){ 
            scanf("%d",&kk); 
            sum+=Del(1,n,1,kk); 
        } 
        printf("Case %d: %lld\n",i,sum); 
    } 
    return 0; 

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#define SIZE 262145
using namespace std;
int array[SIZE<<1];
void creat(int left,int right,int root)
{
    int m;
    array[root] = right-left+1;//每个节点记录包含的数字个数
    if(left == right){
        return;
    }
    m=(right+left)/2;
    creat(left,m,root<<1);      //array[root]的左孩子为array[2*root]
    creat(m+1,right,root<<1|1); //array[root]的右孩子为array[2*root+1]
}
int Del(int left,int right,int root,int key)
{
    int m,flag;
    array[root]--;
    if(left == right){
        return right;
    }
    m=(right+left)/2;
    if(key<=array[root<<1]){//左子树的叶子节点数大于key
        flag=Del(left,m,root<<1,key);
    }
    else{
        key-=array[root<<1];//在右子树中找,key要减去左子树包含的叶子节点数
        flag=Del(m+1,right,root<<1|1,key);
    }
    return flag;
}
int main()
{
    int t,n,k,kk,i,j;
    long long sum;
    scanf("%d",&t);
    for(i=1;i<=t;i++){
        sum=0;
        scanf("%d%d",&n,&k);
        creat(1,n,1);
        for(j=0;j<k;j++){
            scanf("%d",&kk);
            sum+=Del(1,n,1,kk);
        }
        printf("Case %d: %lld\n",i,sum);
    }
    return 0;
}

 

补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,