ACM-ICPC Asia-Chennai Onsite Replay Contest 2017 第三题

题目大意是：给你n个数(x(i))和一个数e，然后问是否存在p(i),使得p(1)x(1)+p(2)x(2)+...+p(n)x(n)=e,且p(1)+...p(n)=1，如果有则输出对应p值，否则输出-1;

思路：我的思路是直接把这个问题看成是解一个方程组即：p(1)x(1)+p(2)x(2)+...+p(n)x(n)=e和p(1)+...p(n)=1。判断其有解还是无解。然后运用一个vector模拟矩阵，不断化简，最后得出解。感觉思路挺清晰的，但是tle，不知道是不是哪里写错了，在这里先暂时分享一下，希望有大神可以看出错误。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<list>
#include<map>
#include<cmath>
using namespace std;
const int maxn=1e3+50;
const double eps=1e-6;
int matrix[2][maxn];
int t,n,e,x,nowsize,st,lo,ans3[maxn];
bool vis[maxn];

struct data{
    int loc;
    double num;
    data(double nn,int lo):num(nn),loc(lo){};
};

int main(){
    while(scanf("%d",&t)!=EOF){
        for(int i=0;i<t;i++){
            list<data> matrix;
            memset(vis,0,sizeof(vis));
            scanf("%d%d",&n,&e);//1 2     3 3
            for(int j=0;j<n;j++){
                scanf("%d",&x);//1
                matrix.push_back(data(x,j));
                ans3[j]=x;
            }
            matrix.push_back(data(e,n));//1 2 3 3
            list<data>::iterator it;
            it=matrix.begin();//
            int k=-(*it).num;//-1
            if(matrix.size()==2)
            if(abs((ans3[0]*1.0)/e-1)>eps){
                printf("-1\n");
                continue;
            }
            while(matrix.size()!=2){///做一次行变换
                for(it=matrix.begin();it!=matrix.end();){
                    (*it).num+=k;
                    if(!(*it).num){
                        matrix.erase(it++);
                        }
                    else it++;
                    }
                    if(matrix.empty())
                        break;
                }
                if(matrix.size()==2){
                    it=matrix.begin();//1
                    double fenmu=(*it).num;//1
                    int loca=(*it).loc;
                    vis[(*it).loc]=true;
                    list<data>::reverse_iterator it;
                    it=matrix.rbegin();
                    double fenzi=(*it).num;//1
                    double ans1=fenzi/fenmu;//1
                    double ans2=1-ans1;//0
                    for(int j=0;j<n;j++){
                        if(vis[j])
                            printf("%.11f",ans1);
                        else if(abs(ans3[j]*ans2+ans1*ans3[loca]-e)<eps){
                            if(ans2==0)
                            printf("0");
                            else
                            printf("%.11f",ans2);
                        }

                        else
                            printf("0");
                        printf("%c",j==n-1?'\n':' ');
                    }
                }
                if(matrix.empty())
                    printf("-1\n");
            }
        }
        return 0;
}

em。。。后来经大佬提醒这是道不折不扣的水题啊！！不过还是一直wa原因竟然是1只能输出1.00，0只能输出0。。。题目也没说清楚啊，icpc也这么坑人的吗。。。不过还是要总结一下，就是对于它给出的数据，无论如何要尊重它原来的格式。

思路：直接找一串数字中的最大值和最小值，然后直接maxvalue*p+(1-p)*minvalue=e，求解这个方程就可以了。当e<minvalue||e>maxvalue时无解。

ac:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<map>
#include<cmath>
#include<iomanip>
using namespace std;
const int maxn=1e3+50;
const double eps=1e-6;
int t,n,x,num[maxn],maxs,mins,maxsloc,minsloc,e;
long double ans1,ans2;
int main(){
    while(cin>>t){
        for(int i=0;i<t;i++){
            memset(num,0,sizeof(num));
            ans1=ans2=0;
            map<int,int> m;
            cin>>n>>e;
            for(int j=0;j<n;j++){
                cin>>num[j];
                m[num[j]]=j;
            }
            if(n==1){
                if(num[0]==e)
                    printf("1.00\n");
                else
                    cout<<-1<<endl;
                continue;
            }
            sort(num,num+n);
            maxs=num[n-1];//4
            maxsloc=m[maxs];//
            mins=num[0];//1
            minsloc=m[mins];
            if(e>maxs||e<mins){
                cout<<-1<<endl;
                continue;
            }
                if(mins==maxs&&mins!=e){
                        cout<<-1<<endl;
                        continue;
                }
                if(maxs==mins)
                    ans2=0;
                else
                ans2=((e-mins)*1.0)/(maxs-mins);
                ans1=1.0-ans2;
                for(int j=0;j<n;j++){
                if(j==minsloc){
                if(abs(ans1-1)<eps)
                printf("1.00");
                else if(ans1==0)
                printf("0");
                else
                cout<<fixed<<setprecision(13)<<ans1;
                }
                else if(j==maxsloc){
                if(abs(ans2-1)<eps)
                printf("1.00");
                else if(ans2==0)
                printf("0");
                else
                cout<<fixed<<setprecision(13)<<ans2;
                }

                else
                    printf("0");
                printf("%c",j==n-1?'\n':' ');
                }
                }
    }
    return 0;
}