XVII Open Cup named after E.V. Pankratiev Stage 14, Grand Prix of Tatarstan, Sunday, April 2, 2017 Problem J. Terminal

题目：Problem J. Terminal
Input file: standard input
Output file: standard input
Time limit: 2 seconds
Memory limit: 256 mebibytes
N programmers from M teams are waiting at the terminal of airport. There are two shuttles at the exit
of terminal, each shuttle can carry no more than K passengers.
Now employees of the airport service need to choose one of the shuttles for each programmer. Note that:
• programmers already formed a queue before assignment to the shuttles;
• each second next programmer in the queue goes to the shuttle he or she is assigned for;
• when all programmers, assigned to the shuttle, are in, shuttle immediately closes door and leaves
the terminal;
• no two programmers from the same team may be assigned to the different shuttles;
• each programmer must be assigned to one of shuttles.
Check if its possible to find such as assignment; if the answer is positive, find minimum sum of waiting
times for each programmer. Waiting time for a person is defined as time when shuttle with this person
left the terminal; it takes one second to programmer to leave the queue and enter the assigned shuttle.
At moment 0 the first programmer begins moving to his shuttle.
Input
First line of the input contains three positive integers N, M and K (M ≤ 2000, M ≤ N ≤ 105, K ≤ 105).
Second line contains description of the queue — N space-separated integers Ai — ids of team of each
programmer in order they are placed in the queue (1 ≤ Ai ≤ M).
Output
If it is impossible to assign programmers to she shuttles following the rules above, print -1. Otherwise
print one integer — minimum sum of waiting times for all programmers.
Examples

standard input	standard input
7 3 5 2 2 1 1 1 3 1	39
12 3 9 1 1 1 2 3 2 2 2 2 2 2 2	116
2 1 2 1 1	4

思路：

　　　如果存在可行解，那么最后一个人一定会上车，不如直接选定上第二辆车，所以第二辆车是第ｎ秒开的。

　　然后枚举上第一辆车的最后一个队的最后一个人是什么时候上车的。

　　怎么判断可行呢？０１背包即可。

 #include <bits/stdc++.h>

 using namespace std;

 #define MP make_pair
 #define PB push_back
 typedef long long LL;
 typedef pair<int,int> PII;
 const double eps=1e-;
 const double pi=acos(-1.0);
 const int K=1e6+;
 const int mod=1e9+;

 int n,m,k;
 LL ls[],rd[],cnt[];
 LL ans=2e18;
 bool dp[][];
 bool cmp(int a,int b)
 {
     return ls[a]<ls[b];
 }
 int main(void)
 {
     //freopen("in.txt","r",stdin);
     scanf("%d%d%d",&n,&m,&k);
     for(int i=;i<=m;i++) rd[i]=i;
     for(int i=,x;i<=n;i++)
         scanf("%d",&x),ls[x]=i,cnt[x]++;
     sort(rd+,rd+m+,cmp);
     dp[][]=;
     for(int i=,now=,pre=;i<=m;i++)
     {
         //printf("%d\n",rd[i]);
         for(int j=;j<=k;j++)
         if(dp[pre][j]&&j+cnt[rd[i]]<=k&&n-j-cnt[rd[i]]<=k)
             ans=min(ans,1LL*(j+cnt[rd[i]])*ls[rd[i]]+1LL*(n-j-cnt[rd[i]])*n);
         for(int j=;j<=k;j++)
         {
             dp[now][j]|=dp[pre][j];
             if(j+cnt[rd[i]]<=k)
                dp[now][j+cnt[rd[i]]]|=dp[pre][j];
             dp[pre][j]=;
         }
         swap(now,pre);
     }
     if(ans==2e18) printf("-1\n");
     else printf("%lld\n",ans);
     return ;
 }