Milking cows
Iahub helps his grandfather at the farm. Today he must milk the cows. There are n cows sitting in a row, numbered from 1 to n from left to right. Each cow is either facing to the left or facing to the right. When Iahub milks a cow, all the cows that see the current cow get scared and lose one unit of the quantity of milk that they can give. A cow facing left sees all the cows with lower indices than her index, and a cow facing right sees all the cows with higher indices than her index. A cow that got scared once can get scared again (and lose one more unit of milk). A cow that has been milked once cannot get scared and lose any more milk. You can assume that a cow never loses all the milk she can give (a cow gives an infinitely amount of milk).
Iahub can decide the order in which he milks the cows. But he must milk each cow exactly once. Iahub wants to lose as little milk as possible. Print the minimum amount of milk that is lost.
Input
The first line contains an integer n (1 ≤ n ≤ 200000). The second line contains n integers a1, a2, ..., an, where ai is 0 if the cow numberi is facing left, and 1 if it is facing right.
Output
Print a single integer, the minimum amount of lost milk.
Please, do not write the %lld specifier to read or write 64-bit integers in С++. It is preferred to use the cin, cout streams or the%I64d specifier.
Sample test(s)
input
4 0 0 1 0
output
1
input
5 1 0 1 0 1
output
3
Note
In the first sample Iahub milks the cows in the following order: cow 3, cow 4, cow 2, cow 1. When he milks cow 3, cow 4 loses 1 unit of milk. After that, no more milk is lost.
##########################editorial#############################################
this is clear that '0 ' can be affected by all '1' left to it .. and all '1' can be affected by all '0' right to it so if array contain some initial zero than we can discard it also if array contain some terminal 1's(ie ones at the end of array ) than we can alsodiscard it ..
ex 00000110101011111 this can be transform to 1101010..
now one more considration is that all if count affect on 1 than we need not to count affect on 0 since only in pair of cow if one affect other than other can affect that one (since cow which milk first will affact other ).. so we need to consider only one type either affect on 1 or on 0 ..
if we count no of 0 right ward each 1 than we can give this ans in 0(n) time ..
since we just need to know no of 0 right to each 1 ..
########################### dp code ######################################
#include<iostream>
using namespace std;
int main()
{
int n;
cin>>n;
int arr[n+1000];
for(int i=0;i<n;i++) cin>>arr[i];
int start=0;
while(arr[start]!=1) start++; eliminating starting 0
int end=n-1;
while(arr[end]!=0) end--;// eliminating ending 1
int dp[n+1000];
dp[end+1]=0;
for(int i=end;i>=0;i--)
{
dp[i]=dp[i+1];
if(arr[i]==0) dp[i]+=1;
}
long long int ans=0;
for(int i=start;i<=end;i++)
{
if(arr[i]==1)
{
ans+=dp[i+1];
}
}
cout<<ans<<endl;
}
################################simple implimentation code ###################
#include<bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); long long int n,i,x,ans=0,ctx=0; cin>>n; for(i=0;i<n;i++) { cin>>x; if(x) ctx++; else ans+=ctx; } cout<<ans<<endl; return 0; }
No comments:
Post a Comment