LOJ P10249 weight 题解

每日一题 day58 打卡

Analysis

这道题搜索的想法非常巧妙，从两端向中间找，这样可以保证仅仅对于head或tail而言，需要用到的前缀和与后缀和是单调递增的，这样排个序就解决了。

值得一提的是，在搜索时开两个变量记录前缀与后缀和，以便计算。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define int long long
 #define maxn 2000+10
 #define rep(i,s,e) for(register int i=s;i<=e;++i)
 #define dwn(i,s,e) for(register int i=s;i>=e;--i)
 using namespace std;
 inline int read()
 {
     int x=,f=;
     char c=getchar();
     while(c<''||c>'') {if(c=='-') f=-; c=getchar();}
     while(c>=''&&c<='') {x=x*+c-''; c=getchar();}
     return f*x;
 }
 inline void write(int x)
 {
     if(x<) {putchar('-'); x=-x;}
     if(x>) write(x/);
     putchar(x%+'');
 }
 int n,m;
 int a[maxn],ans[maxn];
 int book[maxn*];
 int flag=;
 void dfs(int num,int head,int tail,int sum_head,int sum_tail)
 {
     if(flag==) return;
     if(head==tail)
     {
         int x=a[*n]-sum_head-sum_tail;
         if(x>=&&x<=&&book[x]==)
         {
             ans[head]=x;
             rep(i,,n) {write(ans[i]); putchar(' ');}
             flag=;
         }
     }
     else
     {
         int x=a[num]-sum_head;
         if(x>=&&x<=&&book[x]==)
         {
             ans[head]=x;
             dfs(num+,head+,tail,sum_head+x,sum_tail);
             ans[head]=;
         }
         x=a[num]-sum_tail;
         if(x>=&&x<=&&book[x]==)
         {
             ans[tail]=x;
             dfs(num+,head,tail-,sum_head,sum_tail+x);
             ans[tail]=;
         }
     }
 }
 signed main()
 {
     n=read();
     rep(i,,*n) a[i]=read();
     sort(a+,a+*n+);
     m=read();
     rep(i,,m)
     {
         int x=read();
         book[x]=;
     }
     dfs(,,n,,);
     return ;
 }

请各位大佬斧正（反正我不认识斧正是什么意思）