C. Baby Ehab Partitions Again

C. Baby Ehab Partitions Again

题目链接

题目大意
给你一个长度为n的序列要求你使得这个序列无论怎么划分成两个子序列,其和都不会相等。要求你输出这个序列中需要删除元素的下标。

思路

当原序列的和为奇数时不需要删除,这个序列无论怎么划分都不会出现相等的情况,如果为偶数,先除去序列元素的最大公因子,这样方便找出一个奇数,直接去除,然后就是判断这些元素中能不能凑出总和的一半,能不能凑出一半,等价于01背包问题,背包的限制就是总和的一半,最大价值也就是总和的一半。

通过代码

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")
using namespace std;
#define ll long long
#define sl(n) scanf("%lld",&n)
#define pl(n) printf("%lld",n)
#define sdf(n) scanf("%lf",&n)
#define pdf(n) printf("%.lf",n)
#define pE printf("\n")
#define ull unsigned long long
#define pb push_back
#define debug(a) cout<<a<<"??"
#define me(a)  memset(a,0,sizeof(a))
#define pre(beg,en) for(ll i=beg;i<=en;i++)
#define rep(beg,en) for(ll i=beg;i>=en;i--)
#define ph push
#define pi pair<ll,ll>
#define fi first
#define se second
const ll mod = 1e9 + 7;
ll a[3000];
ll dp[200010];
int main(){
  
      ll n,sum=0,sum1=0;
      sl(n);
      pre(1,n)sl(a[i]);

      ll temp=a[1];
      pre(2,n)temp=__gcd(temp,a[i]);
      pre(1,n){
        a[i]/=temp;
        if(a[i]%2)sum1++;
        sum+=a[i];
      }
      if(sum%2){
        cout<<'0';
        return 0;
      }
      ll m=sum/2;
      for(ll i=1;i<=n;i++){
          for(ll j=m;j>=a[i];j--)dp[j]=max(dp[j],dp[j-a[i]]+a[i]);}
          

          if(dp[m]==m){
            puts("1");
            pre(1,n){
                if(a[i]%2){cout<<i;
                return 0;}
      }
          }
          else cout<<"0";

    
      
      return 0;
}
上一篇:前端的心路历程


下一篇:D. Ehab the Xorcist