北理工.程序设计方法实践.软工

北理工.程序设计方法实践.热身-4

百丽宫小菜鸡算法学习的自我记录，希望大家指正与建议。

题干

园长大人看到小白兔善于(找人帮忙)思考，于是给小白兔出了一道题。

有n堆有序的胡萝卜堆，假设第i堆胡萝卜的重量全部相等，为a[i]，并且每一堆胡萝卜的数量也相等，为x。

现在，园长大人希望小白兔确定一个正整数x，使得每堆胡萝卜的总重量的方差 最接近于一个正整数k。

因为小白兔是一只善于(找人帮忙)思考的小白兔，于是他又请你帮忙写一个程序计算符合要求的x

输入描述

第一行两个数n,k 表示有n堆胡萝卜，方差最接近的正整数k

第二行n个数，第i个数表示a[i]

n,a[i] -> [1,10^3]

k -> [1,10^9]

输出描述

输出为一行

若存在符合的x，则输出x，有多个x时，输出最小的那个

若不存在符合的x，则输出 No answer!

个人思路

1. 反解出x的表达式：$x=\sqrt{\frac{k}{S^2}}$,再浮点数向下取整，向上取整，比较两者到k的距离，即为答案。
2. 遍历求到k附近的两个坐标

代码更迭

想法一

#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#define N 1005
long long k;
int n, a[N];
double averange(int a[],int n){//求平均值
    double sum = 0;
    for(int i = 0;i < n;i++){
        sum += a[i];
    }
    return sum/n;
}
double var(int a[],int n){//求方差
    double m = averange(a,n);
    double temp = 0;
    for(int i = 0;i < n;i++){
        temp += pow(a[i]-m,2);
    }
    return temp/(n-1);
}
int move(long double num){//就近取整
    long long out = floor(num);
    if((num - out)>0.5)out = ceil(num);
    return out;
}
int main()
{
    scanf("%d%lld",&n,&k);
    for(int i = 0;i < n;i++){
        scanf("%d",&a[i]);
    }
    double temp = var(a,n);
    double x = sqrt(k/temp); //找到中心点
    int out;
    int out1 = ceil(x);//向上取整
    int out2 = floor(x);//向下取整
    long double out11 = pow(out1,2)*temp;
    long double out22 = pow(out2,2)*temp;
    if(abs(move(out11)-k) > abs(move(out22)-k))out = out1;
    else out = out2;//比较两者的距离，取就近的一个
    if(out > 0 && out < 10000)printf("%d\n",out);
    else printf("No answer!\n");//我的垃圾判断初版
}

程序运行的结果并不理想。WA好几个。
原因如下：

1. 何时判断no answer：方差为0的时候
2. 两者就近距离有无特殊例外：向下取整为0怎么解决
   于是经过无数次的提交试错，在学长的突然一句发问
   我悟了。

#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#define N 1005
long long k;
int n, a[N];
double averange(int a[],int n){//求平均值
    long double sum = 0;
    for(int i = 0;i < n;i++){
        sum += a[i];
    }
    return sum/n;
}
long double var(int a[],int n){//求方差
    double m = averange(a,n);
    long double temp = 0;
    for(int i = 0;i < n;i++){
        temp += pow(a[i]-m,2);
    }
    return temp/n;
}
int main()
{
    scanf("%d%lld",&n,&k);
    for(int i = 0;i < n;i++){
        scanf("%d",&a[i]);
    }
    long double temp = var(a,n);
    if(temp==0)printf("No answer!\n");//方差为0，直接开摆
    else {
        long double x = sqrt(k/temp); 
        //printf("%lf\n",x);
        long long left = floor(x);
        long long right = ceil(x);
        if(fabs(pow(left,2)*temp-k) < fabs(pow(right,2)*temp-k)){//左边的差小于右边
            if(left == 0)printf("%lld\n",right);//向下取整为0，需要取右边的
            else printf("%lld\n",left);
        }
        else printf("%lld\n",right);  
    }
}

一整天的debug终于过了，焯！！！！！

想法二

直接上一个比较奇特的解法，判断条件有些奇怪，面向答案猜过程的产物，大佬同学提供的思路。

#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#define N 1005
long long k;
int n, a[N];
double averange(int a[],int n){//求平均值
    long double sum = 0;
    for(int i = 0;i < n;i++){
        sum += a[i];
    }
    return sum/n;
}
long double var(int a[],int n){//求方差
    double m = averange(a,n);
    long double temp = 0;
    for(int i = 0;i < n;i++){
        temp += pow(a[i]-m,2);
    }
    return temp/n;
}

int main()
{
    scanf("%d%lld",&n,&k);
    //getchar();
    for(int i = 0;i < n;i++){
        scanf("%d",&a[i]);
    }
    long double temp = var(a,n);
    if(temp == 0)printf("No answer!\n");
    else {
        long long point;
        for(point = 1;pow((double)point,2)*temp<k;point++);
        long long left = point - 1;//k左边第一个点
        long long right = point;//k右边第一个点
        long double disr = fabs(pow((double)right,2)*temp-k);//到k的距离
        long double disl = fabs(pow((double)left,2)*temp-k);//到k的距离
    
        if(disl <= disr && pow(left,2)*temp > 10000)printf("%lld\n",left);//神奇的判断条件
        else printf("%lld\n",right);
    }
}

当然最后的判断思路按照想法一也可以做出。
另外的提醒：

真是收获良多的一天啊（薅头发）。